implement type checking functions

1 files changed, 26 insertions, 0 deletions

diff --git a/neb/interpreter.py b/neb/interpreter.py
index e64e420..2461f90 100644
--- a/neb/interpreter.py
+++ b/neb/interpreter.py
@@ -820,3 +820,29 @@ def interpretTypeOf(symbol, args, env, ns):
     return Type(f"{args[0].type_}")
 
 GLOBALS.register("typeof", Builtin(interpretTypeOf, [Arg("candidate", TypeEnum.ANY)]))
+
+def interpretIsString(symbol, args, env, ns):
+    return Bool(isinstance(args[0], String))
+
+GLOBALS.register("string?", Builtin(interpretIsString, [Arg("arg", TypeEnum.ANY)]))
+
+def interpretIsInt(symbol, args, env, ns):
+    return Bool(isinstance(args[0], Int))
+
+GLOBALS.register("int?", Builtin(interpretIsInt, [Arg("arg", TypeEnum.ANY)]))
+
+def interpretIsFloat(symbol, args, env, ns):
+    return Bool(isinstance(args[0], Float))
+
+GLOBALS.register("float?", Builtin(interpretIsFloat, [Arg("arg", TypeEnum.ANY)]))
+
+def interpretIsNumber(symbol, args, env, ns):
+    ret = isinstance(args[0], Int) or isinstance(args[0], Float)
+    return Bool(ret)
+
+GLOBALS.register("number?", Builtin(interpretIsNumber, [Arg("arg", TypeEnum.ANY)]))
+
+def interpretIsBool(symbol, args, env, ns):
+    return Bool(isinstance(args[0], Bool))
+
+GLOBALS.register("bool?", Builtin(interpretIsBool, [Arg("arg", TypeEnum.ANY)]))