Remove numpy test dep and rework matrix multiplication tests.

ionelmc · ionelmc · commit 13f1a557022f · 2023-01-04T19:52:05.000+02:00
diff --git a/tests/test_lazy_object_proxy.py b/tests/test_lazy_object_proxy.py
@@ -913,23 +913,35 @@ def test_mul(lop):
 
 
 def test_matmul(lop):
-    import numpy
+    class MatmulClass:
+        def __init__(self, value):
+            self.value = value
 
-    one = numpy.array((1, 2, 3))
-    two = numpy.array((2, 3, 4))
-    assert one @ two == 20
+        def __matmul__(self, other):
+            return self.value * other.value
 
-    one = lop.Proxy(lambda: numpy.array((1, 2, 3)))
-    two = lop.Proxy(lambda: numpy.array((2, 3, 4)))
-    assert one @ two == 20
+        def __rmatmul__(self, other):
+            return other + self.value
 
-    one = lop.Proxy(lambda: numpy.array((1, 2, 3)))
-    two = numpy.array((2, 3, 4))
-    assert one @ two == 20
+    one = MatmulClass(123)
+    two = MatmulClass(234)
+    assert one @ two == 28782
 
-    one = numpy.array((1, 2, 3))
-    two = lop.Proxy(lambda: numpy.array((2, 3, 4)))
-    assert one @ two == 20
+    one = lop.Proxy(lambda: MatmulClass(123))
+    two = lop.Proxy(lambda: MatmulClass(234))
+    assert one @ two == 28782
+
+    one = lop.Proxy(lambda: MatmulClass(123))
+    two = MatmulClass(234)
+    assert one @ two == 28782
+
+    one = 123
+    two = lop.Proxy(lambda: MatmulClass(234))
+    assert one @ two == 357
+
+    one = lop.Proxy(lambda: 123)
+    two = lop.Proxy(lambda: MatmulClass(234))
+    assert one @ two == 357
 
 
 def test_div(lop):
diff --git a/tox.ini b/tox.ini
@@ -42,7 +42,6 @@ deps =
     pytest
     pytest-benchmark
     Django
-    numpy
     objproxies==0.9.4
     hunter
     cover: pytest-cov
