add more test for setitem

2021-07-17 19:55:53 +08:00 · 2021-07-17 19:55:53 +08:00 · 242084349f
parent b4cb572c90
commit 242084349f
1 changed files with 5 additions and 0 deletions
--- a/python/jittor/test/test_setitem.py
+++ b/python/jittor/test/test_setitem.py
@ -78,6 +78,11 @@ class TestSetitem(unittest.TestCase):
        arr4_res.data[0,0,1,1] = 1
        assert arr4[0,0,1,1] != 1

+        arr4 = jt.random((3,3))
+        arr4_res = arr4[...,:,:2]
+        arr4_res.data[1,1] = 1
+        assert arr4[1,1] != 1
+
        arr5 = jt.random((4,2,3,3))
        arr5_res = arr5[1:3,:,:,:]
        arr5_res.data[1,0,1,1] = 1