[Dy2St] Use np.testing.assert_allclose instead of equal check in dy2st uts (#72326)

SigureMo · web-flow · commit c4900c503674 · 2025-04-17T12:16:09.000+08:00
diff --git a/test/dygraph_to_static/test_ast_util.py b/test/dygraph_to_static/test_ast_util.py
@@ -66,7 +66,7 @@ def test_ast2func_dygraph(self):
             x_v = paddle.to_tensor(x_data)
             true_ret = func(x_v).numpy()
             test_ret = self._ast2func(func)(x_v).numpy()
-            self.assertTrue((true_ret == test_ret).all())
+            np.testing.assert_allclose(true_ret, test_ret)
 
     @test_ast_only
     @test_pir_only
@@ -85,7 +85,7 @@ def func(x):
                 test_ret = self._ast2func(func)(x_v)
                 exe = paddle.static.Executor(paddle.CPUPlace())
                 ret = exe.run(main_program, fetch_list=[true_ret, test_ret])
-                self.assertTrue((ret[0] == ret[1]).all())
+                np.testing.assert_allclose(ret[0], ret[1])
 
     @test_ast_only
     @test_pir_only
diff --git a/test/dygraph_to_static/test_dict.py b/test/dygraph_to_static/test_dict.py
@@ -139,7 +139,7 @@ def train(self):
             return ret.numpy()
 
     def test_ast_to_func(self):
-        self.assertTrue((self._run_dygraph() == self._run_static()).all())
+        np.testing.assert_allclose(self._run_dygraph(), self._run_static())
 
 
 # Tests for dict pop
@@ -227,13 +227,7 @@ def train(self):
             return ret.numpy()
 
     def test_ast_to_func(self):
-        dygraph_result = self._run_dygraph()
-        static_result = self._run_static()
-
-        self.assertTrue(
-            (dygraph_result == static_result).all(),
-            msg=f"dygraph result: {dygraph_result}\nstatic result: {static_result}",
-        )
+        np.testing.assert_allclose(self._run_dygraph(), self._run_static())
 
 
 class TestDictCmpInFor(Dy2StTestBase):
diff --git a/test/dygraph_to_static/test_lambda.py b/test/dygraph_to_static/test_lambda.py
@@ -99,23 +99,23 @@ def run_dygraph(self, func, to_static=False):
 
     def test_call_lambda_as_func(self):
         fn = call_lambda_as_func
-        self.assertTrue((self.run_dygraph(fn) == self.run_static(fn)).all())
+        np.testing.assert_allclose(self.run_dygraph(fn), self.run_static(fn))
 
     def test_call_lambda_directly(self):
         fn = call_lambda_directly
-        self.assertTrue((self.run_dygraph(fn) == self.run_static(fn)).all())
+        np.testing.assert_allclose(self.run_dygraph(fn), self.run_static(fn))
 
     def test_call_lambda_in_func(self):
         fn = call_lambda_in_func
-        self.assertTrue((self.run_dygraph(fn) == self.run_static(fn)).all())
+        np.testing.assert_allclose(self.run_dygraph(fn), self.run_static(fn))
 
     def test_call_lambda_with_if_expr(self):
         fn = call_lambda_with_if_expr
-        self.assertTrue((self.run_dygraph(fn) == self.run_static(fn)).all())
+        np.testing.assert_allclose(self.run_dygraph(fn), self.run_static(fn))
 
     def test_call_lambda_with_if_expr2(self):
         fn = call_lambda_with_if_expr2
-        self.assertTrue((self.run_dygraph(fn) == self.run_static(fn)).all())
+        np.testing.assert_allclose(self.run_dygraph(fn), self.run_static(fn))
 
 
 if __name__ == '__main__':
diff --git a/test/dygraph_to_static/test_typing.py b/test/dygraph_to_static/test_typing.py
@@ -109,7 +109,7 @@ def build_net(self):
 
     def run_dy(self):
         out, np_data = self.net(self.x)
-        self.assertTrue(np.equal(np_data, np.ones_like(np_data)).all())
+        np.testing.assert_allclose(np_data, np.ones_like(np_data))
         return out
 
 
