real-python-drf/puppy_store/puppies/tests/test_models.py

from django.test import TestCase
from ..models import Puppy

class PuppyTest(TestCase):
    """
    Test module for Puppy model
    """

    def setUp(self) -> None:
        Puppy.objects.create(
            name='Casper',
            age=3,
            breed='Bull Dog',
            color='Black'
        )
        Puppy.objects.create(
            name='Muffin',
            age=1,
            breed='Gradane',
            color='Brown'
        )

    def test_puppy_get_breed(self):
        puppy_casper = Puppy.objects.get(name='Casper')
        puppy_muffin = Puppy.objects.get(name='Muffin')
        self.assertEqual(
            puppy_casper.get_breed(), "Casper belongs to Bull Dog breed."
        )
        self.assertEqual(
            puppy_muffin.get_breed(), "Muffin belongs to Gradane breed."
        )