Add tests for SHAPE.

python-processing-unit · python-processing-unit · commit 90e63514998e · 2026-04-04T22:58:33.000-04:00
diff --git a/tests/cases/failing/shape-arity-two.pre b/tests/cases/failing/shape-arity-two.pre
@@ -0,0 +1 @@
+SHAPE([0d1], [0d2])
diff --git a/tests/cases/failing/shape-arity-zero.pre b/tests/cases/failing/shape-arity-zero.pre
@@ -0,0 +1 @@
+SHAPE()
diff --git a/tests/cases/failing/shape-bool.pre b/tests/cases/failing/shape-bool.pre
@@ -0,0 +1 @@
+SHAPE(TRUE)
diff --git a/tests/cases/failing/shape-flt.pre b/tests/cases/failing/shape-flt.pre
@@ -0,0 +1 @@
+SHAPE(0d1.0)
diff --git a/tests/cases/failing/shape-func.pre b/tests/cases/failing/shape-func.pre
@@ -0,0 +1 @@
+SHAPE(LAMBDA INT: (){})
diff --git a/tests/cases/failing/shape-int.pre b/tests/cases/failing/shape-int.pre
@@ -0,0 +1 @@
+SHAPE(0d1)
diff --git a/tests/cases/failing/shape-map.pre b/tests/cases/failing/shape-map.pre
@@ -0,0 +1 @@
+SHAPE(<>)
diff --git a/tests/cases/failing/shape-str.pre b/tests/cases/failing/shape-str.pre
@@ -0,0 +1 @@
+SHAPE("abc")
diff --git a/tests/cases/failing/shape-thr.pre b/tests/cases/failing/shape-thr.pre
@@ -0,0 +1 @@
+SHAPE(ASYNC{})
diff --git a/tests/cases/passing/shape-rank-1.pre b/tests/cases/passing/shape-rank-1.pre
@@ -0,0 +1,4 @@
+TNS: v = [0d10, 0d20, 0d30, 0d40]
+
+ASSERT(EQ(TYPE(SHAPE(v)), "TNS"))
+ASSERT(EQ(SHAPE(v), [0d4]))
diff --git a/tests/cases/passing/shape-rank-2.pre b/tests/cases/passing/shape-rank-2.pre
@@ -0,0 +1,3 @@
+TNS: m = [[0d1, 0d2, 0d3], [0d4, 0d5, 0d6]]
+
+ASSERT(EQ(SHAPE(m), [0d2, 0d3]))
diff --git a/tests/cases/passing/shape-rank-3.pre b/tests/cases/passing/shape-rank-3.pre
@@ -0,0 +1,3 @@
+TNS: cube = [[[0d1, 0d2]], [[0d3, 0d4]], [[0d5, 0d6]]]
+
+ASSERT(EQ(SHAPE(cube), [0d3, 0d1, 0d2]))
diff --git a/tests/cases/passing/shape-result-is-1d.pre b/tests/cases/passing/shape-result-is-1d.pre
@@ -0,0 +1,8 @@
+TNS: t = [[[0d1], [0d2], [0d3]], [[0d4], [0d5], [0d6]]]
+TNS: dims = SHAPE(t)
+
+ASSERT(EQ(TYPE(dims), "TNS"))
+ASSERT(EQ(TLEN(dims, 0d1), 0d3))
+ASSERT(EQ(dims[0d1], 0d2))
+ASSERT(EQ(dims[0d2], 0d3))
+ASSERT(EQ(dims[0d3], 0d1))
