(self)
| 59 | self.assertEqual(d["label"][1], "spleen_label_31.nii.gz") |
| 60 | |
| 61 | def test_deterministic(self): |
| 62 | set_determinism(0) |
| 63 | res_1 = list(ThreadDataLoader(torch.arange(5), batch_size=2, buffer_size=2, shuffle=True, num_workers=0)) |
| 64 | |
| 65 | set_determinism(0) |
| 66 | num_workers = 2 if sys.platform == "linux" else 1 |
| 67 | res_2 = list( |
| 68 | ThreadDataLoader(torch.arange(5), batch_size=2, buffer_size=3, shuffle=True, num_workers=num_workers) |
| 69 | ) |
| 70 | |
| 71 | set_determinism(None) |
| 72 | assert_allclose(torch.cat(res_1), torch.cat(res_2), type_test=False) |
| 73 | |
| 74 | def test_time(self): |
| 75 | dataset = Dataset(data=self.datalist * 2, transform=self.transform) # contains data for 2 batches |
nothing calls this directly
no test coverage detected