[Dy2St] Use np.testing module to get better error message and tolerance settings in test_ifelse (#71951)

SigureMo · web-flow · commit 3cf2510610d0 · 2025-03-28T14:31:07.000+08:00
diff --git a/test/dygraph_to_static/test_ifelse.py b/test/dygraph_to_static/test_ifelse.py
@@ -95,7 +95,7 @@ def _run_dygraph(self, to_static=False):
         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())
 
 
 class TestDygraphIfElse2(TestDygraphIfElse):
@@ -106,7 +106,9 @@ def setUp(self):
     # TODO(dev): fix AST mode
     @disable_test_case((ToStaticMode.AST, IrMode.PT))
     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(), atol=1e-7, rtol=1e-7
+        )
 
 
 class TestDygraphIfElse3(Dy2StTestBase):
@@ -126,7 +128,7 @@ def _run_dygraph(self, to_static=False):
         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())
 
 
 class TestDygraphIfElse4(TestDygraphIfElse):
@@ -141,7 +143,7 @@ def setUp(self):
         self.dyfunc = dyfunc_with_if_else_with_list_generator
 
     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())
 
 
 class TestDygraphNestedIfElse(Dy2StTestBase):
@@ -161,7 +163,7 @@ def _run_dygraph(self, to_static=False):
         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())
 
 
 class TestDygraphNestedIfElse2(TestDygraphIfElse):
@@ -170,7 +172,7 @@ def setUp(self):
         self.dyfunc = nested_if_else_2
 
     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())
 
 
 class TestDygraphNestedIfElse3(Dy2StTestBase):
@@ -190,7 +192,7 @@ def _run_dygraph(self, to_static=False):
         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())
 
 
 def dyfunc_ifExp_with_while(x):
@@ -302,7 +304,7 @@ def _run_dygraph(self, to_static=False):
         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())
 
 
 class TestDygraphIfElseNet(Dy2StTestBase):
@@ -329,7 +331,7 @@ def _run(self, to_static=False):
             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())
 
 
 # Test to call function ahead caller.
@@ -381,7 +383,9 @@ def setUp(self):
         self.Net = NetWithExternalFunc
 
     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(), rtol=1e-7, atol=1e-8
+        )
 
 
 class DiffModeNet1(paddle.nn.Layer):
@@ -438,19 +442,15 @@ def _run(self, mode, to_static):
             return ret.numpy()
 
     def test_train_mode(self):
-        self.assertTrue(
-            (
-                self._run(mode='train', to_static=True)
-                == self._run(mode='train', to_static=False)
-            ).all()
+        np.testing.assert_allclose(
+            self._run(mode='train', to_static=True),
+            self._run(mode='train', to_static=False),
         )
 
     def test_infer_mode(self):
-        self.assertTrue(
-            (
-                self._run(mode='infer', to_static=True)
-                == self._run(mode='infer', to_static=False)
-            ).all()
+        np.testing.assert_allclose(
+            self._run(mode='infer', to_static=True),
+            self._run(mode='infer', to_static=False),
         )
 
 
