feat: Add equality test for booleans (#394)

Closes #363
CQCL · Aug 21, 2024 · dd702ce · dd702ce
1 parent 0fdbbbb
commit dd702ce
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/guppylang/prelude/builtins.py b/guppylang/prelude/builtins.py
@@ -75,6 +75,9 @@ def __and__(self: bool, other: bool) -> bool: ...
     @guppy.custom(builtins, NoopCompiler())
     def __bool__(self: bool) -> bool: ...
 
+    @guppy.hugr_op(builtins, logic_op("Eq", [tys.TypeArg(tys.BoundedNatArg(n=2))]))
+    def __eq__(self: bool, other: bool) -> bool: ...
+
     @guppy.hugr_op(builtins, int_op("ifrombool"))
     def __int__(self: bool) -> int: ...
 

diff --git a/tests/integration/test_if.py b/tests/integration/test_if.py
@@ -264,3 +264,13 @@ def foo(x: int) -> int:
         return z
 
     validate(foo)
+
+
+def test_eq(validate):
+    @compile_guppy
+    def foo() -> bool:
+        x = True
+        y = not x
+        return x == y
+
+    validate(foo)