add two initial asserts
@@ -53,8 +53,10 @@ class ParamStoreDictTests(TestCase):
assert not modules_are_equal()
pyro.module("mymodule", self.linear_module3, update_module_params=False)
+ assert id(self.linear_module3.weight) != id(pyro.param('mymodule$$$weight'))
assert not modules_are_equal()
pyro.module("mymodule", self.linear_module3, update_module_params=True)
+ assert id(self.linear_module3.weight) == id(pyro.param('mymodule$$$weight'))
assert modules_are_equal()
myparam = pyro.param("myparam")
