From ae1ee24178ecd53ac5ef3fc2055e4be8b9602ac9 Mon Sep 17 00:00:00 2001
From: Sarah Boyce <42296566+sarahboyce@users.noreply.github.com>
Date: Tue, 17 Sep 2024 12:40:21 +0200
Subject: Fixed #35766 -- Handled slices in BaseChoiceIterator.

---
 tests/model_fields/tests.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'tests/model_fields')

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, [])
-- 
cgit v1.3