quyuan 1 년 전
부모
커밋
34461d486e
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  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)