1
from __future__ import absolute_import
3
from django.test import TestCase
5
from .models import Person
8
class ChoicesTests(TestCase):
9
def test_display(self):
10
a = Person.objects.create(name='Adrian', gender='M')
11
s = Person.objects.create(name='Sara', gender='F')
12
self.assertEqual(a.gender, 'M')
13
self.assertEqual(s.gender, 'F')
15
self.assertEqual(a.get_gender_display(), 'Male')
16
self.assertEqual(s.get_gender_display(), 'Female')
18
# If the value for the field doesn't correspond to a valid choice,
19
# the value itself is provided as a display value.
21
self.assertEqual(a.get_gender_display(), '')
24
self.assertEqual(a.get_gender_display(), 'U')