Expand H3 ref2v slots and tag routing

tests/test_dumas_h3_longvideos.py

@@ -132,6 +132,51 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
         fn(spoken)
         self.assertGreaterEqual(fn.cache_info().hits, 2)
 
+    def test_resolve_tagged_refs_preserves_sparse_socket_numbers(self):
+        refs = [None, "img2", None, None, None, None, "img7", None, "img9"]
+
+        text, images, dropped = self.module.resolve_tagged_refs(
+            "Mara <Picture 7> turns toward Jon <Picture 2> while <Picture 9> watches.",
+            refs,
+        )
+
+        self.assertEqual(
+            text,
+            "Mara <Picture 2> turns toward Jon <Picture 1> while <Picture 3> watches.",
+        )
+        self.assertEqual(images, ["img2", "img7", "img9"])
+        self.assertEqual(dropped, [])
+
+    def test_resolve_tagged_refs_drops_unconnected_sparse_slots(self):
+        refs = [None, "img2", None, None, None, None, "img7", None, None]
+
+        text, images, dropped = self.module.resolve_tagged_refs(
+            "Use <Picture 7>, skip <Picture 4>, keep <Picture 2>.",
+            refs,
+        )
+
+        self.assertEqual(text, "Use <Picture 2>, skip, keep <Picture 1>.")
+        self.assertEqual(images, ["img2", "img7"])
+        self.assertEqual(dropped, [4])
+
+    def test_shot_references_uses_all_connected_sparse_slots(self):
+        refs = [None, "img2", None, "img4", None, None, "img7", None, None]
+
+        self.assertEqual(
+            self.module.shot_references(refs, "first shot", 0, None),
+            ["img2", "img4", "img7"],
+        )
+        self.assertEqual(
+            self.module.shot_references(refs, "every shot", 3, None),
+            ["img2", "img4", "img7"],
+        )
+
+    def test_input_types_expose_nine_ref_slots(self):
+        optional = self.module.H3LongVideos.INPUT_TYPES()["optional"]
+
+        for index in range(1, 10):
+            self.assertIn(f"ref_image_{index}", optional)
+
 
 if __name__ == "__main__":
     unittest.main()