pokeapi/pokemon_v2/test_models.py

12 lines
336 B
Python

from django.test import TestCase
from pokemon_v2.models import *
class AbilityTestCase(TestCase):
def setUp(self):
Ability.objects.create(name="Smell", generation_id=3, is_main_series=True)
def fields_are_valid(self):
smell = Ability.objects.get(name="Smell")
self.assertEqual(smell.generation_id, 3)