lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add more test for setitem

This commit is contained in:
Dun Liang 2021-07-17 19:55:53 +08:00
parent b4cb572c90
commit 242084349f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/jittor/test/test_setitem.py
View File

@ -78,6 +78,11 @@ class TestSetitem(unittest.TestCase):
arr4_res.data[0,0,1,1] = 1 arr4_res.data[0,0,1,1] = 1
assert arr4[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 = jt.random((4,2,3,3))
arr5_res = arr5[1:3,:,:,:] arr5_res = arr5[1:3,:,:,:]
arr5_res.data[1,0,1,1] = 1 arr5_res.data[1,0,1,1] = 1