fix test directory migration

PaddlePaddle · Oct 13, 2020 · d1cb0c4 · d1cb0c4
1 parent c425cf1
commit d1cb0c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_directory_migration.py b/python/paddle/fluid/tests/unittests/test_directory_migration.py
@@ -88,8 +88,8 @@ def test_new_directory(self):
             stderr=subprocess.PIPE)
         stdout, stderr = ps_proc.communicate()
 
-        assert "Error" not in str(stderr), "Error: Can't" \
-            " import Module {}".format(module)
+        assert "Error" not in str(stderr), "ErrorMessage:\n{}".format(
+            bytes.decode(stderr))
 
     def test_old_directory(self):
         old_directory = [