summaryrefslogtreecommitdiff
path: root/tests/model_fields
diff options
context:
space:
mode:
authorSarah Boyce <42296566+sarahboyce@users.noreply.github.com>2024-09-17 12:40:21 +0200
committerSarah Boyce <42296566+sarahboyce@users.noreply.github.com>2024-09-18 16:45:53 +0200
commitae1ee24178ecd53ac5ef3fc2055e4be8b9602ac9 (patch)
tree1412f68d1d4999797e56ad394ccbb92c266b893c /tests/model_fields
parent9ca1f6eff6f19d1ae074d289c6c4209073351805 (diff)
Fixed #35766 -- Handled slices in BaseChoiceIterator.
Diffstat (limited to 'tests/model_fields')
-rw-r--r--tests/model_fields/tests.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/tests/model_fields/tests.py b/tests/model_fields/tests.py
index 36e54d4b8b..3d856d36c5 100644
--- a/tests/model_fields/tests.py
+++ b/tests/model_fields/tests.py
@@ -183,6 +183,33 @@ class ChoicesTests(SimpleTestCase):
self.choices_from_callable.choices.func(), [(0, "0"), (1, "1"), (2, "2")]
)
+ def test_choices_slice(self):
+ for choices, expected_slice in [
+ (self.empty_choices.choices, []),
+ (self.empty_choices_bool.choices, []),
+ (self.empty_choices_text.choices, []),
+ (self.with_choices.choices, [(1, "A")]),
+ (self.with_choices_dict.choices, [(1, "A")]),
+ (self.with_choices_nested_dict.choices, [("Thing", [(1, "A")])]),
+ (self.choices_from_iterator.choices, [(0, "0"), (1, "1")]),
+ (self.choices_from_callable.choices.func(), [(0, "0"), (1, "1")]),
+ (self.choices_from_callable.choices, [(0, "0"), (1, "1")]),
+ ]:
+ with self.subTest(choices=choices):
+ self.assertEqual(choices[:2], expected_slice)
+
+ def test_choices_negative_index(self):
+ for choices, expected_choice in [
+ (self.with_choices.choices, (1, "A")),
+ (self.with_choices_dict.choices, (1, "A")),
+ (self.with_choices_nested_dict.choices, ("Thing", [(1, "A")])),
+ (self.choices_from_iterator.choices, (2, "2")),
+ (self.choices_from_callable.choices.func(), (2, "2")),
+ (self.choices_from_callable.choices, (2, "2")),
+ ]:
+ with self.subTest(choices=choices):
+ self.assertEqual(choices[-1], expected_choice)
+
def test_flatchoices(self):
self.assertEqual(self.no_choices.flatchoices, [])
self.assertEqual(self.empty_choices.flatchoices, [])