Parcourir la source

test: batch process demo PDFs- Update test block to iterate through multiple demo PDF files

- Use os.path.join to construct file paths for better cross-platform compatibility
- Remove hardcoded file path
myhloli il y a 11 mois
Parent
commit
e11e6b3255
2 fichiers modifiés avec 5 ajouts et 2 suppressions
  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)

Fichier diff supprimé car celui-ci est trop grand
+ 0 - 0
demo/small_ocr.json


Certains fichiers n'ont pas été affichés car il y a eu trop de fichiers modifiés dans ce diff