add cli test

opendatalab · Apr 29, 2024 · 80ca01b · 80ca01b
1 parent d61cb49
commit 80ca01b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/tests/test_cli/conf/conf.py b/tests/test_cli/conf/conf.py
@@ -2,7 +2,7 @@
 conf = {
 "code_path": os.environ.get('GITHUB_WORKSPACE'),
 "pdf_dev_path" : os.environ.get('GITHUB_WORKSPACE') + "/tests/test_cli/pdf_dev",
-"pdf_res_path": "/home/quyuan/code/Magic-PDF/Magic-PDF/Magic-PDF/data"
+"pdf_res_path": "/home/quyuan/code/Magic-PDF/Magic-PDF/Magic-PDF/ci"
 
 }
 
diff --git a/tests/test_cli/lib/common.py b/tests/test_cli/lib/common.py
@@ -11,8 +11,14 @@ def count_folders_and_check_contents(directory):
     for item in contents:
         # 检查是否为文件夹
         if os.path.isdir(os.path.join(directory, item)):
-            folder_count += 1
             # 检查文件夹是否为空
             folder_path = os.path.join(directory, item)
-            assert os.listdir(folder_path) is not None
-    assert folder_count == 3 
+            for folder in os.listdir(folder_path):
+                folder_count = folder_count + 1
+                assert os.listdir(folder_path) is not None
+    print (folder_count)
+    assert folder_count == 13
+
+
+if __name__ == "__main__":
+    count_folders_and_check_contents("/home/quyuan/code/Magic-PDF/Magic-PDF/Magic-PDF/ci")