Merge pull request #232 from OpShin/fix/correct_in_list_check

Fix an error where UnionType was always allowed to be compared with List[X]

opshin/tests/test_misc.py

@@ -2298,3 +2298,19 @@ def validator(x: int) -> str:
         code = compiler.compile(ast).compile()
         res = uplc_eval(uplc.Apply(code, uplc.PlutusInteger(1)))
         self.assertEqual(res.value, b"hello")
+
+    @unittest.expectedFailure
+    def test_in_list(self):
+        source_code = """
+from opshin.prelude import *
+
+def validator(
+    d: Nothing,
+    r: Nothing,
+    context: ScriptContext,
+):
+    assert context.purpose in context.tx_info.signatories
+"""
+        ast = compiler.parse(source_code)
+        code = compiler.compile(ast)
+        code = code.compile()

opshin/types.py

@@ -579,7 +579,7 @@ def cmp(self, op: cmpop, o: "Type") -> plt.AST:
         if (
             isinstance(o, ListType)
             and isinstance(o.typ, InstanceType)
-            and (o.typ.typ >= t or t >= o.typ.typ for t in self.typs)
+            and any(o.typ.typ >= t or t >= o.typ.typ for t in self.typs)
         ):
             if isinstance(op, In):
                 return plt.Lambda(