Create test_math.py

adtzlr · Dec 6, 2022 · 8b007c4 · 8b007c4
1 parent 57717b2
commit 8b007c4
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/test_math.py b/tests/test_math.py
@@ -0,0 +1,21 @@
+import tensortrax as tr
+import tensortrax.math as tm
+import numpy as np
+
+
+def test_math():
+
+    F = np.ones((3, 3))
+    T = tr.Tensor(F)
+
+    assert isinstance(F @ T, tr.Tensor)
+    assert isinstance(T @ F, tr.Tensor)
+    assert isinstance(T * F, tr.Tensor)
+    assert isinstance(F * T, tr.Tensor)
+
+    assert isinstance(F / T, tr.Tensor)
+    assert isinstance(F * T, tr.Tensor)
+
+
+if __name__ == "__main__":
+    test_math()