common.py 801 Bytes
Newer Older
quyuan's avatar
quyuan committed
1
import subprocess
quyuan's avatar
quyuan committed
2
import os
quyuan's avatar
quyuan committed
3
def check_shell(cmd):
quyuan's avatar
quyuan committed
4
    res = os.system(cmd)
quyuan's avatar
quyuan committed
5
6
7
8
9
10
11
12
13
14
15
    assert res == 0

def count_folders_and_check_contents(directory):
    # 获取目录下的所有文件和文件夹
    contents = os.listdir(directory)
    folder_count = 0
    for item in contents:
        # 检查是否为文件夹
        if os.path.isdir(os.path.join(directory, item)):
            # 检查文件夹是否为空
            folder_path = os.path.join(directory, item)
quyuan's avatar
quyuan committed
16
17
18
19
20
21
22
23
24
            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")