(self, workers)
| 79 | |
| 80 | @parameterized.expand([[1], [0]]) |
| 81 | def test_randomize(self, workers): |
| 82 | set_determinism(0) |
| 83 | dataset = _RandomDataset() |
| 84 | dataloader = DataLoader(dataset, batch_size=2, shuffle=True, num_workers=workers) |
| 85 | output = [] |
| 86 | for _ in range(1): # need persistent workers for reproducibility of num_workers 0, 1 |
| 87 | for batch in dataloader: |
| 88 | output.extend(batch.data.numpy().flatten().tolist()) |
| 89 | set_determinism(None) |
| 90 | self.assertListEqual(output, [594, 170, 292, 589, 153, 811, 21, 550]) |
| 91 | |
| 92 | def test_zipdataset(self): |
| 93 | dataset = ZipDataset([_RandomDataset(), ZipDataset([_RandomDataset(), _RandomDataset()])]) |
nothing calls this directly
no test coverage detected