Просмотр исходного кода

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 месяцев назад
Родитель
Сommit
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


Некоторые файлы не были показаны из-за большого количества измененных файлов