common.py 599 B

1234567891011121314151617
  1. import subprocess
  2. def check_shell(cmd):
  3. res = subprocess.check_output(cmd, shell=True)
  4. assert res == 0
  5. def count_folders_and_check_contents(directory):
  6. # 获取目录下的所有文件和文件夹
  7. contents = os.listdir(directory)
  8. folder_count = 0
  9. for item in contents:
  10. # 检查是否为文件夹
  11. if os.path.isdir(os.path.join(directory, item)):
  12. folder_count += 1
  13. # 检查文件夹是否为空
  14. folder_path = os.path.join(directory, item)
  15. assert os.listdir(folder_path) is not None
  16. assert folder_count == 3