(self)
| 113 | previous_batch = None |
| 114 | |
| 115 | def test_thread_workers(self): |
| 116 | dataset = Dataset(data=self.datalist, transform=self.transform) |
| 117 | dataloader = ThreadDataLoader(dataset=dataset, batch_size=2, num_workers=2, use_thread_workers=True) |
| 118 | |
| 119 | for d in dataloader: |
| 120 | self.assertEqual(d["image"][0], "spleen_19.nii.gz") |
| 121 | self.assertEqual(d["image"][1], "spleen_31.nii.gz") |
| 122 | self.assertEqual(d["label"][0], "spleen_label_19.nii.gz") |
| 123 | self.assertEqual(d["label"][1], "spleen_label_31.nii.gz") |
| 124 | |
| 125 | |
| 126 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected