Skip to content
Snippets Groups Projects
Select Git revision
  • main default protected
  • robertos-branch
  • aifmd_dataset_support
  • v1.3.1
  • v1.3.0
  • v1.2.0
  • v1.1.0
  • v1.0.2
  • v1.0.1
  • v1.0.0
  • v1.0.0b0
  • v1.0.0a0
12 results
Blame Permalink
test_montgomery_shenzhen_indian_padchest.py 2.42 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72 






# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""Tests for the aggregated Montgomery-Shenzhen-Indian-PadChest dataset."""

import importlib

import pytest


@pytest.mark.parametrize(
    "name,padchest_name",
    [
        ("default", "tb_idiap"),
    ],
)
def test_split_consistency(name: str, padchest_name: str):
    montgomery = importlib.import_module(
        f".{name}", "mednet.config.data.montgomery"
    ).datamodule

    shenzhen = importlib.import_module(
        f".{name}", "mednet.config.data.shenzhen"
    ).datamodule

    indian = importlib.import_module(
        f".{name}", "mednet.config.data.indian"
    ).datamodule

    padchest = importlib.import_module(
        f".{padchest_name}", "mednet.config.data.padchest"
    ).datamodule

    combined = importlib.import_module(
        f".{name}", "mednet.config.data.montgomery_shenzhen_indian_padchest"
    ).datamodule

    MontgomeryLoader = importlib.import_module(
        ".datamodule", "mednet.config.data.montgomery"
    ).RawDataLoader

    ShenzhenLoader = importlib.import_module(
        ".datamodule", "mednet.config.data.shenzhen"
    ).RawDataLoader

    IndianLoader = importlib.import_module(
        ".datamodule", "mednet.config.data.indian"
    ).RawDataLoader

    PadChestLoader = importlib.import_module(
        ".datamodule", "mednet.config.data.padchest"
    ).RawDataLoader

    for split in ("train", "validation", "test"):
        assert montgomery.splits[split][0][0] == combined.splits[split][0][0]
        assert isinstance(montgomery.splits[split][0][1], MontgomeryLoader)
        assert isinstance(combined.splits[split][0][1], MontgomeryLoader)

        assert shenzhen.splits[split][0][0] == combined.splits[split][1][0]
        assert isinstance(shenzhen.splits[split][0][1], ShenzhenLoader)
        assert isinstance(combined.splits[split][1][1], ShenzhenLoader)

        assert indian.splits[split][0][0] == combined.splits[split][2][0]
        assert isinstance(indian.splits[split][0][1], IndianLoader)
        assert isinstance(combined.splits[split][2][1], IndianLoader)

        if split != "validation":
            # padchest has no validation
            assert padchest.splits[split][0][0] == combined.splits[split][3][0]
            assert isinstance(padchest.splits[split][0][1], PadChestLoader)
            assert isinstance(combined.splits[split][3][1], PadChestLoader)