Add tests for TNS-TNS arithmetic.

python-processing-unit · python-processing-unit · commit 7a3ec940c910 · 2026-04-05T01:45:41.000-04:00
diff --git a/tests/cases/failing/madd-arity-one.pre b/tests/cases/failing/madd-arity-one.pre
@@ -0,0 +1 @@
+MADD([0d1])
diff --git a/tests/cases/failing/madd-arity-three.pre b/tests/cases/failing/madd-arity-three.pre
@@ -0,0 +1 @@
+MADD([0d1], [0d2], [0d3])
diff --git a/tests/cases/failing/madd-arity-zero.pre b/tests/cases/failing/madd-arity-zero.pre
@@ -0,0 +1 @@
+MADD()
diff --git a/tests/cases/failing/madd-first-non-tns.pre b/tests/cases/failing/madd-first-non-tns.pre
@@ -0,0 +1 @@
+MADD(0d1, [0d2])
diff --git a/tests/cases/failing/madd-non-numeric.pre b/tests/cases/failing/madd-non-numeric.pre
@@ -0,0 +1 @@
+MADD(["a"], ["b"])
diff --git a/tests/cases/failing/madd-second-non-tns.pre b/tests/cases/failing/madd-second-non-tns.pre
@@ -0,0 +1 @@
+MADD([0d1], 0d2)
diff --git a/tests/cases/failing/madd-shape-mismatch.pre b/tests/cases/failing/madd-shape-mismatch.pre
@@ -0,0 +1 @@
+MADD([0d1, 0d2], [0d3])
diff --git a/tests/cases/failing/mdiv-arity-one.pre b/tests/cases/failing/mdiv-arity-one.pre
@@ -0,0 +1 @@
+MDIV([0d1])
diff --git a/tests/cases/failing/mdiv-arity-three.pre b/tests/cases/failing/mdiv-arity-three.pre
@@ -0,0 +1 @@
+MDIV([0d1], [0d2], [0d3])
diff --git a/tests/cases/failing/mdiv-arity-zero.pre b/tests/cases/failing/mdiv-arity-zero.pre
@@ -0,0 +1 @@
+MDIV()
diff --git a/tests/cases/failing/mdiv-first-non-tns.pre b/tests/cases/failing/mdiv-first-non-tns.pre
@@ -0,0 +1 @@
+MDIV(0d1, [0d2])
diff --git a/tests/cases/failing/mdiv-non-numeric.pre b/tests/cases/failing/mdiv-non-numeric.pre
@@ -0,0 +1 @@
+MDIV(["a"], ["b"])
diff --git a/tests/cases/failing/mdiv-second-non-tns.pre b/tests/cases/failing/mdiv-second-non-tns.pre
@@ -0,0 +1 @@
+MDIV([0d1], 0d2)
diff --git a/tests/cases/failing/mdiv-shape-mismatch.pre b/tests/cases/failing/mdiv-shape-mismatch.pre
@@ -0,0 +1 @@
+MDIV([0d1, 0d2], [0d3])
diff --git a/tests/cases/failing/mdiv-zero.pre b/tests/cases/failing/mdiv-zero.pre
@@ -0,0 +1 @@
+MDIV([0d1, 0d2], [0d1, 0d0])
diff --git a/tests/cases/failing/mmul-arity-one.pre b/tests/cases/failing/mmul-arity-one.pre
@@ -0,0 +1 @@
+MMUL([0d1])
diff --git a/tests/cases/failing/mmul-arity-three.pre b/tests/cases/failing/mmul-arity-three.pre
@@ -0,0 +1 @@
+MMUL([0d1], [0d2], [0d3])
diff --git a/tests/cases/failing/mmul-arity-zero.pre b/tests/cases/failing/mmul-arity-zero.pre
@@ -0,0 +1 @@
+MMUL()
diff --git a/tests/cases/failing/mmul-first-non-tns.pre b/tests/cases/failing/mmul-first-non-tns.pre
@@ -0,0 +1 @@
+MMUL(0d1, [0d2])
diff --git a/tests/cases/failing/mmul-non-numeric.pre b/tests/cases/failing/mmul-non-numeric.pre
@@ -0,0 +1 @@
+MMUL(["a"], ["b"])
diff --git a/tests/cases/failing/mmul-second-non-tns.pre b/tests/cases/failing/mmul-second-non-tns.pre
@@ -0,0 +1 @@
+MMUL([0d1], 0d2)
diff --git a/tests/cases/failing/mmul-shape-mismatch.pre b/tests/cases/failing/mmul-shape-mismatch.pre
@@ -0,0 +1 @@
+MMUL([0d1, 0d2], [0d3])
diff --git a/tests/cases/failing/mprod-arity-zero.pre b/tests/cases/failing/mprod-arity-zero.pre
@@ -0,0 +1 @@
+MPROD()
diff --git a/tests/cases/failing/mprod-non-numeric.pre b/tests/cases/failing/mprod-non-numeric.pre
@@ -0,0 +1 @@
+MPROD(["a"], ["b"])
diff --git a/tests/cases/failing/mprod-non-tns-arg.pre b/tests/cases/failing/mprod-non-tns-arg.pre
@@ -0,0 +1 @@
+MPROD([0d1], 0d2)
diff --git a/tests/cases/failing/mprod-shape-mismatch.pre b/tests/cases/failing/mprod-shape-mismatch.pre
@@ -0,0 +1 @@
+MPROD([0d1, 0d2], [0d3])
diff --git a/tests/cases/failing/msum-arity-zero.pre b/tests/cases/failing/msum-arity-zero.pre
@@ -0,0 +1 @@
+MSUM()
diff --git a/tests/cases/failing/msum-non-numeric.pre b/tests/cases/failing/msum-non-numeric.pre
@@ -0,0 +1 @@
+MSUM(["a"], ["b"])
diff --git a/tests/cases/failing/msum-non-tns-arg.pre b/tests/cases/failing/msum-non-tns-arg.pre
@@ -0,0 +1 @@
+MSUM([0d1], 0d2)
diff --git a/tests/cases/failing/msum-shape-mismatch.pre b/tests/cases/failing/msum-shape-mismatch.pre
@@ -0,0 +1 @@
+MSUM([0d1, 0d2], [0d3])
diff --git a/tests/cases/passing/madd-basic.pre b/tests/cases/passing/madd-basic.pre
@@ -0,0 +1,4 @@
+TNS: out = MADD([0d1, 0d2, 0d3], [0d4, 0d5, 0d6])
+
+ASSERT(EQ(TYPE(out), "TNS"))
+ASSERT(EQ(out, [0d5, 0d7, 0d9]))
diff --git a/tests/cases/passing/madd-rank2-shape.pre b/tests/cases/passing/madd-rank2-shape.pre
@@ -0,0 +1,6 @@
+TNS: a = [[0d1, 0d2], [0d3, 0d4]]
+TNS: b = [[0d10, 0d20], [0d30, 0d40]]
+TNS: out = MADD(a, b)
+
+ASSERT(EQ(out, [[0d11, 0d22], [0d33, 0d44]]))
+ASSERT(EQ(SHAPE(out), [0d2, 0d2]))
diff --git a/tests/cases/passing/mdiv-basic.pre b/tests/cases/passing/mdiv-basic.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MDIV([0d8, 0d6, 0d4], [0d2, 0d3, 0d2]), [0d4, 0d2, 0d2]))
diff --git a/tests/cases/passing/mdiv-rank2.pre b/tests/cases/passing/mdiv-rank2.pre
@@ -0,0 +1,4 @@
+TNS: out = MDIV([[0d9, 0d8], [0d6, 0d4]], [[0d3, 0d2], [0d3, 0d2]])
+
+ASSERT(EQ(out, [[0d3, 0d4], [0d2, 0d2]]))
+ASSERT(EQ(SHAPE(out), [0d2, 0d2]))
diff --git a/tests/cases/passing/mmul-basic.pre b/tests/cases/passing/mmul-basic.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MMUL([0d1, 0d2, 0d3], [0d4, 0d5, 0d6]), [0d4, 0d10, 0d18]))
diff --git a/tests/cases/passing/mmul-rank2-shape.pre b/tests/cases/passing/mmul-rank2-shape.pre
@@ -0,0 +1,6 @@
+TNS: a = [[0d1, 0d2], [0d3, 0d4]]
+TNS: b = [[0d5, 0d6], [0d7, 0d8]]
+TNS: out = MMUL(a, b)
+
+ASSERT(EQ(out, [[0d5, 0d12], [0d21, 0d32]]))
+ASSERT(EQ(SHAPE(out), [0d2, 0d2]))
diff --git a/tests/cases/passing/mprod-single.pre b/tests/cases/passing/mprod-single.pre
@@ -0,0 +1,3 @@
+TNS: x = [[0d2, 0d3], [0d4, 0d5]]
+
+ASSERT(EQ(MPROD(x), x))
diff --git a/tests/cases/passing/mprod-three.pre b/tests/cases/passing/mprod-three.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MPROD([0d2, 0d3], [0d4, 0d5], [0d6, 0d7]), [0d48, 0d105]))
diff --git a/tests/cases/passing/mprod-two.pre b/tests/cases/passing/mprod-two.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MPROD([0d2, 0d3], [0d4, 0d5]), [0d8, 0d15]))
diff --git a/tests/cases/passing/msum-single.pre b/tests/cases/passing/msum-single.pre
@@ -0,0 +1,3 @@
+TNS: x = [[0d1, 0d2], [0d3, 0d4]]
+
+ASSERT(EQ(MSUM(x), x))
diff --git a/tests/cases/passing/msum-three.pre b/tests/cases/passing/msum-three.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MSUM([0d1, 0d2], [0d3, 0d4], [0d5, 0d6]), [0d9, 0d12]))
diff --git a/tests/cases/passing/msum-two.pre b/tests/cases/passing/msum-two.pre
@@ -0,0 +1 @@
+ASSERT(EQ(MSUM([0d1, 0d2], [0d3, 0d4]), [0d4, 0d6]))
