common.py 583 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
16
17
18
    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_count += 1
            # 检查文件夹是否为空
            folder_path = os.path.join(directory, item)
            assert os.listdir(folder_path) is not None
    assert folder_count == 3