(self)
| 22 | class TestDatasetFunc(unittest.TestCase): |
| 23 | |
| 24 | def test_seg_values(self): |
| 25 | with tempfile.TemporaryDirectory() as tempdir: |
| 26 | # prepare test datalist file |
| 27 | test_data = { |
| 28 | "name": "Spleen", |
| 29 | "description": "Spleen Segmentation", |
| 30 | "labels": {"0": "background", "1": "spleen"}, |
| 31 | "training": [ |
| 32 | {"image": "spleen_19.nii.gz", "label": "spleen_19.nii.gz"}, |
| 33 | {"image": "spleen_31.nii.gz", "label": "spleen_31.nii.gz"}, |
| 34 | ], |
| 35 | "test": ["spleen_15.nii.gz", "spleen_23.nii.gz"], |
| 36 | } |
| 37 | json_str = json.dumps(test_data) |
| 38 | file_path = os.path.join(tempdir, "test_data.json") |
| 39 | with open(file_path, "w") as json_file: |
| 40 | json_file.write(json_str) |
| 41 | |
| 42 | data_list = DatasetFunc( |
| 43 | data=file_path, func=load_decathlon_datalist, data_list_key="training", base_dir=tempdir |
| 44 | ) |
| 45 | # partition dataset for train / validation |
| 46 | data_partition = DatasetFunc( |
| 47 | data=data_list, func=lambda x, **kwargs: partition_dataset(x, **kwargs)[0], num_partitions=2 |
| 48 | ) |
| 49 | dataset = Dataset(data=data_partition, transform=None) |
| 50 | self.assertEqual(dataset[0]["image"], os.path.join(tempdir, "spleen_19.nii.gz")) |
| 51 | self.assertEqual(dataset[0]["label"], os.path.join(tempdir, "spleen_19.nii.gz")) |
| 52 | |
| 53 | |
| 54 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected