[lang-ref] ( list_files_in_dir ) ( python )

@pytest.fixture
def pushd(tmp_path):
    old = os.getcwd()
    os.chdir(tmp_path)
    yield
    os.chdir(old)

def test_list_files_in_dir(pushd):
    # os.listdir()
    os.makedirs('dir1/sub1', exist_ok=True)

    paths = [
        'dir1/file1.txt',
        'dir1/file2.txt',
        'dir1/file3.txt',
        'dir1/sub1/file11.txt',
        'dir1/sub1/file12.txt',
    ]
    for path in paths:
        Path(path).touch()

    results = os.listdir('dir1')
    assert sorted(results) == ['file1.txt', 'file2.txt', 'file3.txt', 'sub1']