Code Duplication    Length = 14-17 lines in 2 locations

test/unit/test_nifti_loader.py 1 location

@@ 14-30 (lines=17) @@
11
12
13
def get_loader(loader_name):
14
    if loader_name in [
15
        "paired",
16
        "unpaired",
17
        "grouped",
18
    ]:
19
        dir_paths = [f"./data/test/nifti/{loader_name}/test"]
20
        name = "fixed_images" if loader_name == "paired" else "images"
21
        grouped = loader_name == "grouped"
22
    elif loader_name == "multi_dirs_grouped":
23
        dir_paths = [
24
            "./data/test/nifti/grouped/train",
25
            "./data/test/nifti/grouped/test",
26
        ]
27
        name = "images"
28
        grouped = True
29
    else:
30
        raise ValueError
31
    loader = NiftiFileLoader(dir_paths=dir_paths, name=name, grouped=grouped)
32
    return loader
33

test/unit/test_h5_loader.py 1 location

@@ 20-33 (lines=14) @@
17
18
19
def get_loader(loader_name):
20
    if loader_name in [
21
        "paired",
22
        "unpaired",
23
        "grouped",
24
    ]:
25
        dir_paths = [f"./data/test/h5/{loader_name}/test"]
26
        name = "fixed_images" if loader_name == "paired" else "images"
27
        grouped = loader_name == "grouped"
28
    elif loader_name == "multi_dirs_grouped":
29
        dir_paths = ["./data/test/h5/grouped/train", "./data/test/h5/grouped/test"]
30
        name = "images"
31
        grouped = True
32
    else:
33
        raise ValueError
34
    loader = H5FileLoader(dir_paths=dir_paths, name=name, grouped=grouped)
35
    return loader
36