Add tests for CONV.

python-processing-unit · python-processing-unit · commit 62861a95eb1b · 2026-04-05T01:23:05.000-04:00
diff --git a/tests/cases/failing/conv-arity-eight.pre b/tests/cases/failing/conv-arity-eight.pre
@@ -0,0 +1 @@
+CONV([0d1], [0d1], 0d1, 0d1, 0d0, 0d0, [], 0d9)
diff --git a/tests/cases/failing/conv-arity-one.pre b/tests/cases/failing/conv-arity-one.pre
@@ -0,0 +1 @@
+CONV([0d1])
diff --git a/tests/cases/failing/conv-arity-zero.pre b/tests/cases/failing/conv-arity-zero.pre
@@ -0,0 +1 @@
+CONV()
diff --git a/tests/cases/failing/conv-extended-bias-not-tns.pre b/tests/cases/failing/conv-extended-bias-not-tns.pre
@@ -0,0 +1 @@
+CONV([[[0d1]]], [[[[0d1]]]], bias = 0d1)
diff --git a/tests/cases/failing/conv-extended-bias-size-mismatch.pre b/tests/cases/failing/conv-extended-bias-size-mismatch.pre
@@ -0,0 +1 @@
+CONV([[[0d1]]], [[[[0d1, 0d1]]]], bias = [0d5])
diff --git a/tests/cases/failing/conv-extended-kernel-not-rank4.pre b/tests/cases/failing/conv-extended-kernel-not-rank4.pre
@@ -0,0 +1 @@
+CONV([[[0d1]]], [[[0d1]]], stride_w = 0d1)
diff --git a/tests/cases/failing/conv-extended-stride-w-non-int.pre b/tests/cases/failing/conv-extended-stride-w-non-int.pre
@@ -0,0 +1 @@
+CONV([[[0d1]]], [[[[0d1]]]], stride_w = "1")
diff --git a/tests/cases/failing/conv-extended-x-not-rank3.pre b/tests/cases/failing/conv-extended-x-not-rank3.pre
@@ -0,0 +1 @@
+CONV([[0d1]], [[[[0d1]]]], stride_w = 0d1)
diff --git a/tests/cases/failing/conv-first-non-tns.pre b/tests/cases/failing/conv-first-non-tns.pre
@@ -0,0 +1 @@
+CONV(0d1, [0d1])
diff --git a/tests/cases/failing/conv-legacy-even-kernel.pre b/tests/cases/failing/conv-legacy-even-kernel.pre
@@ -0,0 +1 @@
+CONV([0d1, 0d2, 0d3], [0d1, 0d1])
diff --git a/tests/cases/failing/conv-legacy-rank-mismatch.pre b/tests/cases/failing/conv-legacy-rank-mismatch.pre
@@ -0,0 +1 @@
+CONV([0d1, 0d2], [[0d1, 0d1, 0d1]])
diff --git a/tests/cases/failing/conv-second-non-tns.pre b/tests/cases/failing/conv-second-non-tns.pre
@@ -0,0 +1 @@
+CONV([0d1], 0d1)
diff --git a/tests/cases/passing/conv-extended-basic.pre b/tests/cases/passing/conv-extended-basic.pre
@@ -0,0 +1,6 @@
+TNS: x = [[[0d1], [0d2], [0d3]], [[0d4], [0d5], [0d6]]]
+TNS: kernel = [[[[0d1, 0d2]]]]
+TNS: out = CONV(x, kernel, stride_w = 0d1)
+
+ASSERT(EQ(SHAPE(out), [0d2, 0d3, 0d2]))
+ASSERT(EQ(out, [[[0d1, 0d2], [0d2, 0d4], [0d3, 0d6]], [[0d4, 0d8], [0d5, 0d10], [0d6, 0d12]]]))
diff --git a/tests/cases/passing/conv-extended-bias.pre b/tests/cases/passing/conv-extended-bias.pre
@@ -0,0 +1,6 @@
+TNS: x = [[[0d3]]]
+TNS: kernel = [[[[0d2, 0d3]]]]
+TNS: out = CONV(x, kernel, bias = [0d10, 0d20])
+
+ASSERT(EQ(SHAPE(out), [0d1, 0d1, 0d2]))
+ASSERT(EQ(out, [[[0d16, 0d29]]]))
diff --git a/tests/cases/passing/conv-extended-flt-output-type.pre b/tests/cases/passing/conv-extended-flt-output-type.pre
@@ -0,0 +1,6 @@
+TNS: x = [[[0d2.0]]]
+TNS: kernel = [[[[0d1.5]]]]
+TNS: out = CONV(x, kernel, stride_w = 0d1)
+
+ASSERT(EQ(out, [[[0d3.0]]]))
+ASSERT(EQ(TYPE(out[0d1, 0d1, 0d1]), "FLT"))
diff --git a/tests/cases/passing/conv-legacy-1d-clamp.pre b/tests/cases/passing/conv-legacy-1d-clamp.pre
@@ -0,0 +1,7 @@
+TNS: x = [0d1, 0d2, 0d3]
+TNS: k = [0d1, 0d1, 0d1]
+TNS: out = CONV(x, k)
+
+ASSERT(EQ(TYPE(out), "TNS"))
+ASSERT(EQ(SHAPE(out), [0d3]))
+ASSERT(EQ(out, [0d4, 0d6, 0d8]))
diff --git a/tests/cases/passing/conv-legacy-flt-output-type.pre b/tests/cases/passing/conv-legacy-flt-output-type.pre
@@ -0,0 +1,4 @@
+TNS: out = CONV([0d2.0], [0d1.0])
+
+ASSERT(EQ(out, [0d2.0]))
+ASSERT(EQ(TYPE(out[0d1]), "FLT"))
diff --git a/tests/cases/passing/conv-legacy-int-output-type.pre b/tests/cases/passing/conv-legacy-int-output-type.pre
@@ -0,0 +1,4 @@
+TNS: out = CONV([0d7], [0d1])
+
+ASSERT(EQ(out, [0d7]))
+ASSERT(EQ(TYPE(out[0d1]), "INT"))
diff --git a/tests/cases/passing/conv-legacy-rank2-shape.pre b/tests/cases/passing/conv-legacy-rank2-shape.pre
@@ -0,0 +1,6 @@
+TNS: x = [[0d1, 0d2], [0d3, 0d4]]
+TNS: k = [[0d0, 0d0, 0d0], [0d0, 0d1, 0d0], [0d0, 0d0, 0d0]]
+TNS: out = CONV(x, k)
+
+ASSERT(EQ(SHAPE(out), [0d2, 0d2]))
+ASSERT(EQ(out, x))

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([0d1], [0d1], 0d1, 0d1, 0d0, 0d0, [], 0d9)`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([[[0d1]]], [[[[0d1]]]], bias = 0d1)`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([[[0d1]]], [[[[0d1, 0d1]]]], bias = [0d5])`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([[[0d1]]], [[[0d1]]], stride_w = 0d1)`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([[[0d1]]], [[[[0d1]]]], stride_w = "1")`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+CONV([[0d1]], [[[[0d1]]]], stride_w = 0d1)`