(self, device, dtype)
| 291 | |
| 292 | @parameterized.expand(TESTS) |
| 293 | def test_dataset(self, device, dtype): |
| 294 | ims = [self.get_im(device=device, dtype=dtype)[0] for _ in range(4)] |
| 295 | ds = Dataset(ims) |
| 296 | for i, im in enumerate(ds): |
| 297 | self.check(im, ims[i], ids=True) |
| 298 | |
| 299 | @parameterized.expand(DTYPES) |
| 300 | def test_dataloader(self, dtype): |