refactor(prototypes): merge casting widgets into Dupre kit

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/gallery-tokens/test_gen_tokens.py b/tests/gallery-tokens/test_gen_tokens.py
index 2968119..afd9db1 100644
--- a/tests/gallery-tokens/test_gen_tokens.py
+++ b/tests/gallery-tokens/test_gen_tokens.py
@@ -95,6 +95,28 @@ class ResolveColor(unittest.TestCase):
             gt.resolve_color(FIXTURE, "nonexistent")
 
 
+class IterSectionItems(unittest.TestCase):
+    def test_preserves_section_and_item_order(self):
+        self.assertEqual(
+            list(gt.iter_section_items(FIXTURE, ("amber", "palette"))),
+            [
+                ("gold", "#e2a038"),
+                ("gold-hi", "#ffbe54"),
+                ("amber-grad-top", "#f2c76a"),
+                ("ground", "#151311"),
+                ("wash", "#2c2f32"),
+                ("slate-hi", "#54677d"),
+            ],
+        )
+
+    def test_missing_and_empty_sections_contribute_no_items(self):
+        tokens = {"palette": {}, "amber": {"gold": "#e2a038"}}
+        self.assertEqual(
+            list(gt.iter_section_items(tokens, ("missing", "palette", "amber"))),
+            [("gold", "#e2a038")],
+        )
+
+
 class EmitWebCss(unittest.TestCase):
     def setUp(self):
         self.css = gt.emit_web_css(FIXTURE)