(self, datasets, transform, expected_output, expected_length)
| 52 | |
| 53 | @parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3, TEST_CASE_4]) |
| 54 | def test_value(self, datasets, transform, expected_output, expected_length): |
| 55 | test_dataset = ZipDataset(datasets=datasets, transform=transform) |
| 56 | self.assertEqual(test_dataset[0], expected_output) |
| 57 | self.assertEqual(len(test_dataset), expected_length) |
| 58 | |
| 59 | def test_slicing(self): |
| 60 | test_dataset = ZipDataset(datasets=[Dataset_(5), Dataset_(5), Dataset_(5)], transform=None) |