[BE] use self.assertEquals instead of str equality in test_zero1.py (#…

…5364)
pytorch · Jul 27, 2023 · 8ada333 · 8ada333
1 parent d09cc69
commit 8ada333
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_zero1.py b/test/test_zero1.py
@@ -33,20 +33,20 @@ def test_zero1(self):
 
     opt1.step()
     opt2.step()
-    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
+    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
 
     s1 = opt1.state_dict()
     s2 = opt2.state_dict()
     opt1.load_state_dict(s1)
     opt2.load_state_dict(s2)
-    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
+    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
 
     # step still runnable
     opt1.step()
     opt2.step()
     opt1.load_state_dict(s1)
     opt2.load_state_dict(s2)
-    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
+    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
 
     # step still runnable
     opt1.step()