Videre

linedance-app/venv/lib/python3.12/site-packages/numba/tests/matmul_usecase.py

@@ -0,0 +1,24 @@
+"""Use cases for testing matmul (@)
+"""
+def matmul_usecase(x, y):
+    return x @ y
+
+def imatmul_usecase(x, y):
+    x @= y
+    return x
+
+class DumbMatrix(object):
+
+    def __init__(self, value):
+        self.value = value
+
+    def __matmul__(self, other):
+        if isinstance(other, DumbMatrix):
+            return DumbMatrix(self.value * other.value)
+        return NotImplemented
+
+    def __imatmul__(self, other):
+        if isinstance(other, DumbMatrix):
+            self.value *= other.value
+            return self
+        return NotImplemented