소스 검색

Merge pull request #1075 from myhloli/dev

test: batch process demo PDFs- Update test block to iterate through multiple demo PDF files
Xiaomeng Zhao 11 달 전
부모
커밋
4f13c28282
2개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 5 2
      demo/magic_pdf_parse_main.py
  2. 0 0
      demo/small_ocr.json

+ 5 - 2
demo/magic_pdf_parse_main.py

@@ -139,5 +139,8 @@ def pdf_parse_main(
 
 # 测试
 if __name__ == '__main__':
-    file_path = r'D:\project\20240617magicpdf\Magic-PDF\demo\demo1.pdf'
-    pdf_parse_main(file_path)
+    current_script_dir = os.path.dirname(os.path.abspath(__file__))
+    demo_names = ['demo1', 'demo2', 'small_ocr']
+    for name in demo_names:
+        file_path = os.path.join(current_script_dir, f'{name}.pdf')
+        pdf_parse_main(file_path)

파일 크기가 너무 크기때문에 변경 상태를 표시하지 않습니다.
+ 0 - 0
demo/small_ocr.json


이 변경점에서 너무 많은 파일들이 변경되어 몇몇 파일들은 표시되지 않았습니다.