Fix test full name usage (#52790)

* test * fix test error * fix test error * fix test error

Fix test full name usage (#52790)
* test * fix test error * fix test error * fix test error
aac8da90 · risemeup1 · GitHub · 3fed97f4 · aac8da90
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

test/dygraph_to_static/test_full_name_usage.py test/dygraph_to_static/test_full_name_usage.py +2 -2

未找到文件。
--- a/test/dygraph_to_static/test_full_name_usage.py
+++ b/test/dygraph_to_static/test_full_name_usage.py
@@ -72,9 +72,9 @@ class TestFullNameDecorator(unittest.TestCase):
            np.testing.assert_allclose(
                decorated_call_decorated(x).numpy(), answer, rtol=1e-05
            )
-            with self.assertRaises(NotImplementedError):
+            with self.assertRaises((NotImplementedError, TypeError)):
                DoubleDecorated().double_decorated_func1(x)
-            with self.assertRaises(NotImplementedError):
+            with self.assertRaises((NotImplementedError, TypeError)):
                DoubleDecorated().double_decorated_func2(x)