diff --git a/tests/fpgadataflow/test_npy2hls.py b/tests/fpgadataflow/test_npy2hls.py
index ef9426abbcc225081a867e39506a7fc531ef9e0a..884c16cc3dcae72352b1f14d20c508f81d2f425c 100644
--- a/tests/fpgadataflow/test_npy2hls.py
+++ b/tests/fpgadataflow/test_npy2hls.py
@@ -75,16 +75,21 @@ g++ -o test_npy2apintstream test.cpp /workspace/cnpy/cnpy.cpp \
     assert success
 
 
+test_shapes = [(1, 2, 4), (1, 1, 64), (2, 64)]
+
+
 def test_npy2apintstream_binary():
-    dt = DataType.BINARY
-    W = cutil.gen_finn_dt_tensor(dt, (1, 2, 4))
-    make_npy2apintstream_testcase(W, dt)
+    for test_shape in test_shapes:
+        dt = DataType.BINARY
+        W = cutil.gen_finn_dt_tensor(dt, test_shape)
+        make_npy2apintstream_testcase(W, dt)
 
 
 def test_npy2apintstream_int2():
-    dt = DataType.INT2
-    W = cutil.gen_finn_dt_tensor(dt, (1, 2, 4))
-    make_npy2apintstream_testcase(W, dt)
+    for test_shape in test_shapes:
+        dt = DataType.INT2
+        W = cutil.gen_finn_dt_tensor(dt, test_shape)
+        make_npy2apintstream_testcase(W, dt)
 
 
 def test_array2hexstring():