23 lines
726 B
Python
23 lines
726 B
Python
from django.test import TestCase
|
|
|
|
from .models import Person
|
|
|
|
|
|
class ChoicesTests(TestCase):
|
|
def test_display(self):
|
|
a = Person.objects.create(name='Adrian', gender='M')
|
|
s = Person.objects.create(name='Sara', gender='F')
|
|
self.assertEqual(a.gender, 'M')
|
|
self.assertEqual(s.gender, 'F')
|
|
|
|
self.assertEqual(a.get_gender_display(), 'Male')
|
|
self.assertEqual(s.get_gender_display(), 'Female')
|
|
|
|
# If the value for the field doesn't correspond to a valid choice,
|
|
# the value itself is provided as a display value.
|
|
a.gender = ''
|
|
self.assertEqual(a.get_gender_display(), '')
|
|
|
|
a.gender = 'U'
|
|
self.assertEqual(a.get_gender_display(), 'U')
|