Add golden case/constr test

OpShin · Jan 2, 2025 · 4eebe5b · 4eebe5b
1 parent 84ed868
commit 4eebe5b
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/tests/case_constr/case_constr.flat b/tests/case_constr/case_constr.flat
@@ -0,0 +1 @@
+4e0101009800600290003003480081
diff --git a/tests/case_constr/case_constr.uplc b/tests/case_constr/case_constr.uplc
@@ -0,0 +1,4 @@
+(program
+  1.1.0
+  (case (constr 0)(constr 0(con integer 0))(constr 1(con integer 1)))
+)
diff --git a/tests/test_misc.py b/tests/test_misc.py
@@ -1752,3 +1752,14 @@ def test_append_plutusdata_list(self):
             r.result,
             PlutusConstr(0, []),
         )
+
+    def test_case_constr_encoding(self):
+        with open(Path(__file__).parent / "case_constr" / "case_constr.uplc", "r") as f:
+            program = f.read()
+        p = parse(program)
+        encoded = flatten(p)
+        with open(Path(__file__).parent / "case_constr" / "case_constr.flat", "r") as f:
+            encoded_ref = bytes.fromhex(f.read())
+        self.assertEqual(encoded, encoded_ref)
+        decoded = unflatten(encoded_ref)
+        self.assertEqual(p, decoded)