add static ndrange test

taichi-dev · Jul 4, 2022 · 6e20127 · 6e20127
1 parent 42a9cae
commit 6e20127
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/python/test_ndrange.py b/tests/python/test_ndrange.py
@@ -303,3 +303,15 @@ def example():
             r"Every argument of ndrange should be an integer scalar or a tuple/list of \(int, int\)"
     ):
         example()
+
+
+@test_utils.test()
+def test_static_ndrange_should_accept_numpy_integer():
+    a, b = np.int64(0), np.int32(10)
+
+    @ti.kernel
+    def example():
+        for i in ti.static(ti.ndrange((a, b))):
+            pass
+
+    example()