Add missing check method

CQCL · Nov 27, 2023 · b9c81ef · b9c81ef
1 parent 3e89ac8
commit b9c81ef
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/guppy/prelude/_internal.py b/guppy/prelude/_internal.py
@@ -196,6 +196,14 @@ def synthesize(self, args: list[ast.expr]) -> tuple[ast.expr, GuppyType]:
         const = with_loc(self.node, ast.Constant(value=is_callable))
         return const, BoolType()
 
+    def check(self, args: list[ast.expr], ty: GuppyType) -> ast.expr:
+        args, _ = self.synthesize(args)
+        if not isinstance(ty, BoolType):
+            raise GuppyTypeError(
+                f"Expected expression of type `{ty}`, got `bool`", self.node
+            )
+        return args
+
 
 class IntTruedivCompiler(CustomCallCompiler):
     """Compiler for the `int.__truediv__` method."""