Add test cases for unsupported SequenceMaps

PiperOrigin-RevId: 582031946

tracr/compiler/test_cases.py

@@ -393,7 +393,8 @@
 ]
 
 
-# Programs using features that are currently not supported by Tracr
+# Programs using features that are currently not supported by Tracr and that
+# cause the compiler to throw NotImplementerError.
 UNSUPPORTED_TEST_CASES = [
     dict(
         testcase_name="numerical_categorical_aggregate",
@@ -450,4 +451,29 @@
         vocab={1, 2, 3},
         max_seq_len=5,
     ),
+    dict(
+        testcase_name="numerical_SequenceMap",
+        program=rasp.numerical(
+            rasp.SequenceMap(
+                lambda x, y: x + y,
+                rasp.numerical(rasp.Map(lambda x: x, rasp.indices)),
+                rasp.numerical(rasp.Map(lambda x: x, rasp.tokens)),
+            )
+        ),
+        vocab={1, 2, 3},
+        max_seq_len=5,
+    ),
+    dict(
+        testcase_name="categorical_LinearSequenceMap",
+        program=rasp.categorical(
+            rasp.LinearSequenceMap(
+                rasp.categorical(rasp.indices),
+                rasp.categorical(rasp.tokens),
+                1,
+                1,
+            )
+        ),
+        vocab={1, 2, 3},
+        max_seq_len=5,
+    ),
 ]