Add forward operator tests (OpObj, int).

christabella · christabella · commit 536c1bdc318a · 2020-12-21T08:23:44.000Z
diff --git a/src/embed_tests/TestOperator.cs b/src/embed_tests/TestOperator.cs
@@ -50,31 +50,58 @@ public OperableObject(int num)
             {
                 return new OperableObject(a.Num - b.Num);
             }
+            public static OperableObject operator -(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num - b);
+            }
 
             public static OperableObject operator *(OperableObject a, OperableObject b)
             {
                 return new OperableObject(a.Num * b.Num);
             }
+            public static OperableObject operator *(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num * b);
+            }
 
             public static OperableObject operator /(OperableObject a, OperableObject b)
             {
                 return new OperableObject(a.Num / b.Num);
             }
+            public static OperableObject operator /(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num / b);
+            }
+
 
             public static OperableObject operator &(OperableObject a, OperableObject b)
             {
                 return new OperableObject(a.Num & b.Num);
             }
+            public static OperableObject operator &(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num & b);
+            }
+
 
             public static OperableObject operator |(OperableObject a, OperableObject b)
             {
                 return new OperableObject(a.Num | b.Num);
             }
+            public static OperableObject operator |(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num | b);
+            }
+
 
             public static OperableObject operator ^(OperableObject a, OperableObject b)
             {
                 return new OperableObject(a.Num ^ b.Num);
             }
+            public static OperableObject operator ^(OperableObject a, int b)
+            {
+                return new OperableObject(a.Num ^ b);
+            }
 
             public static OperableObject operator <<(OperableObject a, int offset)
             {
@@ -134,10 +161,27 @@ public void ForwardOperatorOverloads()
 from {module} import *
 cls = {name}
 a = cls(2)
-b = cls(10)
+b = 10
+c = a + b
+assert c.Num == a.Num + b
 
-c = a + b.Num
-assert c.Num == a.Num + b.Num
+c = a - b
+assert c.Num == a.Num - b
+
+c = a * b
+assert c.Num == a.Num * b
+
+c = a / b
+assert c.Num == a.Num // b
+
+c = a & b
+assert c.Num == a.Num & b
+
+c = a | b
+assert c.Num == a.Num | b
+
+c = a ^ b
+assert c.Num == a.Num ^ b
 ");
         }
         [Test]
