add some more tests for is

Author: yangdanny97

compiler/types/classvaluetype.py

@@ -102,7 +102,7 @@ def getWasmName(self):
         elif self.className == "int":
             return "i64"
         else:
-            raise Exception("TODO")
+            return "i32"
 
     def __str__(self):
         return self.className

compiler/wasm_backend.py

@@ -119,6 +119,7 @@ def Program(self, node: Program):
         self.instr('(import "imports" "logString" (func $log_str (param i32)))')
         self.instr('(import "imports" "assert" (func $assert (param i32)))')
         self.instr('(memory (import "js" "mem") 1)')
+        # initialize all globals to 0 for now, since we don't statically allocate strings or arrays
         for v in var_decls:
             self.instr(f"(global ${v.var.identifier.name} (mut {v.var.t.getWasmName()})")
             self.instr(f"{v.var.t.getWasmName()}.const 0")
@@ -131,9 +132,11 @@ def Program(self, node: Program):
         self.builder.func("main")
         self.defaultToGlobals = True
         self.locals = self.builder.newBlock()
+        # initialize memory counter
         self.instr("i32.const 0") # addr 0
         self.instr("i32.const 8") # store value 8
         self.instr("i32.store")
+        # initialize globals
         for v in var_decls:
             self.visit(v.value)
             self.instr(f"global.set ${v.getIdentifier().name}")
@@ -256,18 +259,19 @@ def BinaryExpr(self, node: BinaryExpr):
             self.instr("i64.lt_s")
             self.instr("i64.eqz")
         elif operator == "==":
-            # TODO: refs
-            if leftType == BoolType():
-                self.instr("i32.eq")
-            else:
+            # TODO: refs, string
+            if leftType == IntType():
                 self.instr("i64.eq")
-        elif operator == "!=":
-            if leftType == BoolType():
-                self.instr("i32.ne")
             else:
+                self.instr("i32.eq")
+        elif operator == "!=":
+            if leftType == IntType():
                 self.instr("i64.ne")
+            else:
+                self.instr("i32.ne")
         elif operator == "is":
-            raise Exception("TODO")
+            # pointer comparisons
+            self.instr("i32.eq")
         # logical operators
         elif operator == "and":
             self.instr("i32.and")

tests/runtime/lists.py

@@ -25,6 +25,10 @@ def getNestedIdx(lst:[[int]], idx:int)->[int]:
 a = []
 b = []
 
+assert b is b
+assert not (a is b)
+assert not (a is None)
+
 assert len(x) == 0
 assert len([]) == 0
 assert len([1, 2, 3]) == 3

tests/runtime/operators.py

@@ -21,7 +21,6 @@
 assert w * x == x
 assert 5 // 2 == y
 assert 5 % 2 == x
-assert z is z
 assert not False
 assert not (w != x)
 assert -x == -1
@@ -32,3 +31,10 @@
 assert True if x != y else False
 assert False if x == y else True
 
+assert z is z
+assert None is None
+assert not (object() is object())
+assert z is None
+z = object()
+assert z is z
+