Add complex predicate

WyattBlue · Sep 29, 2023 · ef08114 · ef08114
1 parent 27e2450
commit ef08114
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/auto_editor/lang/palet.py b/auto_editor/lang/palet.py
@@ -788,6 +788,8 @@ def __init__(
             self.arity = len(parms), len(parms)
 
     def __call__(self, *args: Any) -> Any:
+        check_args(self.name, args, self.arity, self.contracts)
+
         if self.arity[1] is None:
             args = tuple(
                 list(args[: len(self.parms) - 1]) + [list(args[len(self.parms) - 1 :])]
@@ -1407,10 +1409,11 @@ def my_eval(env: Env, node: object) -> Any:
     "number?": is_num,
     "real?": is_real,
     "int?": is_int,
-    "nat?": is_nat,
-    "nat1?": is_nat1,
     "float?": is_float,
     "frac?": is_frac,
+    "complex?": Contract("complex?", lambda v: type(v) is complex),
+    "nat?": is_nat,
+    "nat1?": is_nat1,
     "threshold?": is_threshold,
     "any": any_p,
     "bool?": is_bool,