refine unittest

PaddlePaddle · Jan 26, 2022 · b9af510 · b9af510 · paddle-bot-old · Jan 27, 2022
1 parent 489fb28
commit b9af510
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/test_optimizer.py b/python/paddle/fluid/tests/unittests/test_optimizer.py
@@ -1197,7 +1197,7 @@ def test_with_state_dict(self):
                 out_no_state_dict = self.check_with_opt_state_dict(
                     use_save_load=False)
             self.assertTrue(
-                np.allclose(out_use_state_dict[0], out_no_state_dict[0]))
+                np.array_equal(out_use_state_dict, out_no_state_dict))
 
 
 if __name__ == '__main__':