From 242084349f2cf8c90e048847f6f063548c978119 Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Sat, 17 Jul 2021 19:55:53 +0800 Subject: [PATCH] add more test for setitem --- python/jittor/test/test_setitem.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/python/jittor/test/test_setitem.py b/python/jittor/test/test_setitem.py index 6841d6b6..3ad98990 100644 --- 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