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

add test_modules

This commit is contained in:
Dun Liang 2022-03-22 14:38:50 +08:00
parent 31fbf4af9c
commit d4e8092c93
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/jittor/test/test_core.py
View File

@ -112,10 +112,10 @@ class TestCore(unittest.TestCase):
a.y = jt.Module()
a.a = jt.array([1,2,3])
a.b = jt.array([1,2,3])
assert a._modules.keys() == ["x", "y"]
assert list(a._modules.keys()) == ["x", "y"]
assert a._modules['x'] is a.x
assert a._modules['y'] is a.y
assert a._parameters.keys() == ['a', 'b']
assert list(a._parameters.keys()) == ['a', 'b']
assert a._parameters['a'] is a.a
assert a._parameters['b'] is a.b