From f172dcd5a96265663fc89bb9987f2740539adce5 Mon Sep 17 00:00:00 2001
From: JackCaoG <59073027+JackCaoG@users.noreply.github.com>
Date: Thu, 27 Jul 2023 13:39:39 -0700
Subject: [PATCH] Revert "[BE] use self.assertEquals instead of str equality in
 test_zero1.py (#5364)" (#5366)

This reverts commit 8ada333234e376ed217b1937a0ce2da86b832d70.
---
 test/test_zero1.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test/test_zero1.py b/test/test_zero1.py
index b2f04cfd58c..18b81c6c407 100644
--- a/test/test_zero1.py
+++ b/test/test_zero1.py
@@ -33,20 +33,20 @@ def test_zero1(self):
 
     opt1.step()
     opt2.step()
-    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
+    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
 
     s1 = opt1.state_dict()
     s2 = opt2.state_dict()
     opt1.load_state_dict(s1)
     opt2.load_state_dict(s2)
-    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
+    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
 
     # step still runnable
     opt1.step()
     opt2.step()
     opt1.load_state_dict(s1)
     opt2.load_state_dict(s2)
-    self.assertEquals(opt1.state_dict(), opt2.state_dict()['base'])
+    assert str(opt1.state_dict()) == str(opt2.state_dict()['base'])
 
     # step still runnable
     opt1.step()