def test_diff_dir_file_lt(fs): # pylint: disable=invalid-name """ Should sort nested children before next sibling (sibling file name is lower than dir name) """ file_path_lt_dir = join(HOME, 'a') fs.create_file(file_path_lt_dir) _setup_dir_with_file(fs) result = list_files([HOME, DIR_PATH])['result'] # check that all files are listed assert len(result) == 3 # check that directory is listed first _assert_dir_goes_first(result) # file is listed after the directory assert result[2] == { "name": "a", "path": file_path_lt_dir, "isDir": False, "isFile": True, "level": 0 }
def test_same_dir_files(fs): # pylint: disable=invalid-name """Should sort files in the same dir alphabetically""" file0 = join(HOME, 'foo') file1 = join(HOME, 'bar') file2 = join(HOME, 'baz') fs.create_file(file0) fs.create_file(file1) fs.create_file(file2) result = list_files([HOME])['result'] assert len(result) == 3 assert result[0] == { "name": "bar", "path": file1, "isDir": False, "isFile": True, "level": 0 } assert result[1] == { "name": "baz", "path": file2, "isDir": False, "isFile": True, "level": 0 } assert result[2] == { "name": "foo", "path": file0, "isDir": False, "isFile": True, "level": 0 }
def _load(data_dir): dataset = [] for subdir in list_subdirs(data_dir): files = list_files(join(data_dir, subdir), ext='.ppm') for file_name in files: file_path = join(data_dir, subdir, file_name) dataset.append({'data': skimage_read(file_path), 'label': subdir}) return dataset