示例#1
0
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
    }
示例#2
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
    }
示例#3
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