quyuan 1 år sedan
förälder
incheckning
34461d486e
1 ändrade filer med 4 tillägg och 1 borttagningar
  1. 4 1
      tests/test_cli/test_cli.py

+ 4 - 1
tests/test_cli/test_cli.py

@@ -33,7 +33,10 @@ class TestCli:
             pipe.pipe_classify()
             pipe.pipe_parse()
             md_content = pipe.pipe_mk_markdown(image_dir, drop_mode="none")
-            res_path = os.path.join(pdf_dev_path, "mineru", f"{demo_name}.md")
+            dir_path = os.path.join(pdf_dev_path, "mineru")
+            if not os.path.exists(dir_path):
+                os.makedirs(dir_path, exist_ok=True)
+            res_path = os.path.join(dir_path, f"{demo_name}.md")
             with open(res_path, "w+", encoding="utf-8") as f:
                 f.write(md_content)