(self)
| 30 | |
| 31 | class TestDataLoader(unittest.TestCase): |
| 32 | def test_values(self): |
| 33 | datalist = [ |
| 34 | {"image": "spleen_19.nii.gz", "label": "spleen_label_19.nii.gz"}, |
| 35 | {"image": "spleen_31.nii.gz", "label": "spleen_label_31.nii.gz"}, |
| 36 | ] |
| 37 | transform = Compose( |
| 38 | [ |
| 39 | DataStatsd(keys=["image", "label"], data_shape=False, value_range=False, data_value=True), |
| 40 | SimulateDelayd(keys=["image", "label"], delay_time=0.1), |
| 41 | ] |
| 42 | ) |
| 43 | dataset = CacheDataset(data=datalist, transform=transform, cache_rate=0.5, cache_num=1) |
| 44 | n_workers = 0 if sys.platform == "win32" else 2 |
| 45 | dataloader = DataLoader(dataset=dataset, batch_size=2, num_workers=n_workers) |
| 46 | for d in dataloader: |
| 47 | self.assertEqual(d["image"][0], "spleen_19.nii.gz") |
| 48 | self.assertEqual(d["image"][1], "spleen_31.nii.gz") |
| 49 | self.assertEqual(d["label"][0], "spleen_label_19.nii.gz") |
| 50 | self.assertEqual(d["label"][1], "spleen_label_31.nii.gz") |
| 51 | |
| 52 | @parameterized.expand([TEST_CASE_1, TEST_CASE_2]) |
| 53 | def test_exception(self, datalist): |
nothing calls this directly
no test coverage detected