summaryrefslogtreecommitdiff
path: root/tests/choices/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/choices/tests.py')
-rw-r--r--tests/choices/tests.py25
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')
+