cleanup

builtins.simple

@@ -1,14 +1,26 @@
 object nil:
     1
 
+
 def pass:
     nil
 
+
 object inttrait:
+    def eq(other):
+        self $int_eq(other)
+
+    def leq(other):
+        self $int_leq(other)
+    def geq(other):
+        self $int_geq(other)
+    def greater(other):
+        self $int_greater(other)
+    def less(other):
+        self $int_less(other)
+
     def add(other):
         self $int_add(other)
-
-    # Project: Sugar
     def sub(other):
         self $int_sub(other)
     def mul(other):
@@ -20,18 +32,6 @@ object inttrait:
     def inc:
         self $int_inc
 
-    # Project: Boolean
-    def eq(other):
-        self $int_eq(other)
-    def leq(other):
-        self $int_leq(other)
-    def geq(other):
-        self $int_geq(other)
-    def greater(other):
-        self $int_greater(other)
-    def less(other):
-        self $int_less(other)
-
     def tobool:
         self $int_tobool
     def tostr:
@@ -39,8 +39,11 @@ object inttrait:
     def todouble:
         self $int_todouble
 
-# Project: Boolean
+
 object booltrait:
+    def eq(other):
+        self $bool_eq(other)
+
     def and(other):
         self $bool_and(other)
     def or(other):
@@ -55,10 +58,11 @@ object booltrait:
     def todouble:
         self $bool_todouble
 
-# Project: String
+
 object strtrait:
     def eq(other):
         self $str_eq(other)
+
     def add(other):
         self $str_add(other)
     def rev:
@@ -73,10 +77,11 @@ object strtrait:
     def todouble:
         self $str_todouble
 
-# Project: Double
+
 object doubletrait:
     def eq(other):
         self $double_eq(other)
+
     def add(other):
         self $double_add(other)
     def sub(other):

de.churl.simple/primitives.py

@@ -1,37 +1,36 @@
-registry = {}
+registry = {}  # names are keys
 all_primitives = []
 primitive_number_of_arguments = []
 
 
-def primitive(name, unwrap_spec, wrap_spec):  # decorator arguments
+def primitive(name, unwrap_spec, wrap_spec):  # decorator args
     assert '$' + name not in registry, '${name} already defined'.format(name=name)
     primitive_number_of_arguments.append(len(unwrap_spec) - 1)  # first argument is the receiver
 
     def expose(func):  # decorator
         def unwrapper(w_receiver, args_w, space):
             args = [w_receiver] + args_w
-            if len(args) != len(unwrap_spec):  # check that call args match primitive args
+            if len(args) != len(unwrap_spec):  # call args match primitive args
-                raise TypeError(
+                raise TypeError("Expected {ex} arguments, received {re}.".format(ex=len(unwrap_spec), re=len(args)))
-                    "Expected {ex} arguments, received {re}.".format(ex=len(unwrap_spec), re=len(args)))
 
             unwrapped_args = ()
-            for t, arg in zip(unwrap_spec, args):  # unpack values from simple-objects
+            for t, arg in zip(unwrap_spec, args):  # unpack
-                if t in [int, str, float]:  # Project: Double, String
+                if t in [int, str, float]:
                     unwrapped_args += (arg.value,)
-                elif t is bool:  # Project: Boolean
+                elif t is bool:
-                    unwrapped_args += (bool(arg.value),)  # isn't really necessary because "1 or 0" is also valid
+                    unwrapped_args += (bool(arg.value),)
                 else:
                     unwrapped_args += (arg,)
 
             result = func(*unwrapped_args)  # actual call
 
-            if wrap_spec is int:  # wrap the result
+            if wrap_spec is int:  # wrap
                 return space.newint(result)
-            elif wrap_spec is bool:  # Project: Boolean
+            elif wrap_spec is bool:
                 return space.newbool(result)
-            elif wrap_spec is str:  # Project: String
+            elif wrap_spec is str:
                 return space.newstring(result)
-            elif wrap_spec is float:  # Project: Double
+            elif wrap_spec is float:
                 return space.newdouble(result)
             return result
 
@@ -51,12 +50,37 @@ def get_number_of_arguments_of_primitive(idx):
     return primitive_number_of_arguments[idx]
 
 
+# int
+@primitive("int_eq", [int, int], bool)
+def simple_int_eq(a, b):
+    return a == b
+
+
+@primitive("int_leq", [int, int], bool)
+def simple_int_leq(a, b):
+    return a <= b
+
+
+@primitive("int_geq", [int, int], bool)
+def simple_int_geq(a, b):
+    return a >= b
+
+
+@primitive("int_greater", [int, int], bool)
+def simple_int_greater(a, b):
+    return a > b
+
+
+@primitive("int_less", [int, int], bool)
+def simple_int_less(a, b):
+    return a < b
+
+
 @primitive('int_add', [int, int], int)
 def simple_int_add(a, b):
     return a + b
 
 
-# Project: Sugar
 @primitive("int_sub", [int, int], int)
 def simple_int_subtract(a, b):
     return a - b
@@ -82,23 +106,27 @@ def simple_int_increment(a):
     return a + 1
 
 
-# Project: Boolean, String, Double
 @primitive("int_tobool", [int], bool)
 def simple_int_tobool(a):
     return a
 
 
 @primitive("int_tostr", [int], str)
-def simple_int_tobstr(a):
+def simple_int_tostr(a):
     return a
 
 
 @primitive("int_todouble", [int], float)
-def simple_int_tobdouble(a):
+def simple_int_todouble(a):
     return a
 
 
-# Project: Boolean
+# bool
+@primitive("bool_eq", [bool, bool], bool)
+def simple_bool_eq(a, b):
+    return a is b
+
+
 @primitive("bool_and", [bool, bool], bool)
 def simple_bool_and(a, b):
     return a and b
@@ -129,33 +157,7 @@ def simple_bool_todouble(a):
     return a
 
 
-# bool stuff for int
+# string
-@primitive("int_eq", [int, int], bool)
-def simple_int_eq(a, b):
-    return a == b
-
-
-@primitive("int_leq", [int, int], bool)
-def simple_int_leq(a, b):
-    return a <= b
-
-
-@primitive("int_geq", [int, int], bool)
-def simple_int_geq(a, b):
-    return a >= b
-
-
-@primitive("int_greater", [int, int], bool)
-def simple_int_greater(a, b):
-    return a > b
-
-
-@primitive("int_less", [int, int], bool)
-def simple_int_less(a, b):
-    return a < b
-
-
-# Project: String
 @primitive("str_eq", [str, str], bool)
 def simple_str_eq(a, b):
     return a == b
@@ -191,7 +193,7 @@ def simple_str_todouble(a):
     return a
 
 
-# Project: Double
+# double
 @primitive("double_eq", [float, float], bool)
 def simple_double_eq(a, b):
     return a == b

mytests/test_boolean.py

@@ -1,4 +1,3 @@
-import py
 from rply import Token
 
 from simpleast import Program, ExprStatement, BooleanLiteral, Assignment, ImplicitSelf
@@ -69,6 +68,7 @@ y = 1 leq(2)
 
 def test_boolean_conversion():
     ast = parse("""
+w = 1 tostr
 x = 1 tobool
 y = true toint
 z = false toint

mytests/test_string.py

@@ -1,4 +1,3 @@
-import py
 from rply import Token
 
 from simpleast import Program, ExprStatement, BooleanLiteral, ImplicitSelf, Assignment, StringLiteral