pokeapi/pokemon_v2/test_models.py
2016-03-05 11:40:41 +00:00

17 lines
425 B
Python

from __future__ import unicode_literals
from django.test import TestCase
from pokemon_v2.models import * # NOQA
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)