Add tuple comparison operator tests (2/6)

christabella · christabella · commit 318d7efa3d3c · 2021-01-11T13:40:37.000+02:00
diff --git a/src/embed_tests/TestOperator.cs b/src/embed_tests/TestOperator.cs
@@ -212,6 +212,25 @@ public OperableObject(int num)
                 return (a.Num >= b);
             }
 
+            public static bool operator >=(OperableObject a, PyObject b)
+            {
+                using (Py.GIL())
+                {
+                    // Assuming b is a tuple, take the first element.
+                    int bNum = (int)b.ToArray()[0].AsManagedObject(typeof(int));
+                    return a.Num >= bNum;
+                }
+            }
+            public static bool operator <=(OperableObject a, PyObject b)
+            {
+                using (Py.GIL())
+                {
+                    // Assuming b is a tuple, take the first element.
+                    int bNum = (int)b.ToArray()[0].AsManagedObject(typeof(int));
+                    return a.Num <= bNum;
+                }
+            }
+
             public static bool operator <(int a, OperableObject b)
             {
                 return (a < b.Num);
@@ -391,6 +410,27 @@ public void ForwardOperatorOverloads()
 ");
         }
 
+        [Test]
+        public void TupleComparisonOperatorOverloads()
+        {
+                string name = string.Format("{0}.{1}",
+                typeof(OperableObject).DeclaringType.Name,
+                typeof(OperableObject).Name);
+            string module = MethodBase.GetCurrentMethod().DeclaringType.Namespace;
+                PythonEngine.Exec($@"
+from {module} import *
+cls = {name}
+a = cls(2)
+b = (1, 2)
+
+c = a >= b
+assert c == (a.Num >= b[0])
+
+c = a <= b
+assert c == (a.Num <= b[0])
+");
+        }
+
 
         [Test]
         public void ReverseOperatorOverloads()
