quyuan 1 yıl önce
ebeveyn
işleme
8a8d54360e
2 değiştirilmiş dosya ile 4 ekleme ve 2 silme
  1. 1 1
      tests/test_cli/lib/common.py
  2. 3 1
      tests/test_cli/test_cli.py

+ 1 - 1
tests/test_cli/lib/common.py

@@ -17,7 +17,7 @@ def count_folders_and_check_contents(directory):
                 folder_count = folder_count + 1
                 assert os.listdir(folder_path) is not None
     print (folder_count)
-    assert folder_count == 14
+    assert folder_count > 0
 
 
 if __name__ == "__main__":

+ 3 - 1
tests/test_cli/test_cli.py

@@ -5,6 +5,8 @@ import os
 import json
 from magic_pdf.pipe.UNIPipe import UNIPipe
 from magic_pdf.rw.DiskReaderWriter import DiskReaderWriter
+from lib import common
+
 pdf_res_path = conf.conf["pdf_res_path"]
 code_path = conf.conf["code_path"]
 pdf_dev_path = conf.conf["pdf_dev_path"]
@@ -39,7 +41,7 @@ class TestCli:
             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)
-
+            common.count_folders_and_check_contents(res_path)
         
     # def test_pdf_specify_jsonl(self):
     #     """