diff --git a/python/paddle/fluid/tests/unittests/test_directory_migration.py b/python/paddle/fluid/tests/unittests/test_directory_migration.py
index 9dbbdeee31427e95639e533a3b27cbfeae99b10a..3e0c52b7be81be7abea5e5c971ae6aa6a1149c99 100644
--- a/python/paddle/fluid/tests/unittests/test_directory_migration.py
+++ b/python/paddle/fluid/tests/unittests/test_directory_migration.py
@@ -94,8 +94,8 @@ class TestDirectory(unittest.TestCase):
             stderr=subprocess.PIPE)
         stdout, stderr = ps_proc.communicate()
 
-        assert "Error" not in str(stderr), "Error: Can't" \
-            " import Module {}".format(module)
+        self.assertFalse("Error" in str(stderr),
+                         "ErrorMessage:\n{}".format(bytes.decode(stderr)))
 
     def test_old_directory(self):
         old_directory = [
@@ -177,7 +177,7 @@ if count != {len_old_directory}:
             stderr=subprocess.PIPE)
         stdout, stderr = ps_proc.communicate()
 
-        assert "Error" not in str(stdout), str(stdout)
+        self.assertFalse("Error" in str(stdout), bytes.decode(stdout))
 
 
 if __name__ == '__main__':