def test_only_files(): with TemporaryDirectory(dir=TMP) as dirname: for i in range(5): filename = f"{i}.txt" path = os.path.join(dirname, filename) with open(path, "w") as f: f.write("hello") assert count_dirs_and_files(dirname) == (0, 5)
def test_files_and_dirs(): with TemporaryDirectory(dir=TMP) as dirname: for i in range(10): if i % 2 == 0: target_dir = os.path.join(dirname, str(i)) os.makedirs(target_dir) for j in range(5): filename = f"{j}.txt" path = os.path.join(target_dir, filename) with open(path, "w") as f: f.write("hello") assert count_dirs_and_files(dirname) == (5, 25)
def test_only_dirs(): with TemporaryDirectory(dir=TMP) as dirname: for i in range(5): os.makedirs(os.path.join(dirname, str(i))) assert count_dirs_and_files(dirname) == (5, 0)