diff options
Diffstat (limited to 'tests/choices/tests.py')
-rw-r--r-- | tests/choices/tests.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/choices/tests.py b/tests/choices/tests.py new file mode 100644 index 0000000000..03a7d3340d --- /dev/null +++ b/tests/choices/tests.py @@ -0,0 +1,25 @@ +from __future__ import absolute_import + +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') + |