diff --git a/python/paddle/fluid/tests/unittests/test_dropout_op.py b/python/paddle/fluid/tests/unittests/test_dropout_op.py
index 094b157b5e47e917b8bd7087afe802ef0ce5514f..3128fd4d0406a85a9bacca8460548c4391e22893 100644
--- a/python/paddle/fluid/tests/unittests/test_dropout_op.py
+++ b/python/paddle/fluid/tests/unittests/test_dropout_op.py
@@ -1147,7 +1147,7 @@ class TestDropOutWithProbTensor(unittest.TestCase):
         for x in self.inputs:
             static_res = self.run_static(x)
             dygraph_res = self.run_dygraph(x)
-            self.assertTrue(np.array_equal(static_res, dygraph_res))
+            np.testing.assert_array_equal(static_res, dygraph_res)
 
 
 class TestRandomValue(unittest.TestCase):