def test_cifar100(self, mock_ext_check, mock_int_check): mock_ext_check.return_value = True mock_int_check.return_value = True with cifar_root('CIFAR100') as root: dataset = torchvision.datasets.CIFAR100(root, train=True, download=True) self.generic_classification_dataset_test(dataset) img, target = dataset[0] self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target) dataset = torchvision.datasets.CIFAR100(root, train=False, download=True) self.generic_classification_dataset_test(dataset) img, target = dataset[0] self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target)
def test_cifar100(self, mock_ext_check, mock_int_check): mock_ext_check.return_value = True mock_int_check.return_value = True with cifar_root('CIFAR100') as root: dataset = torchvision.datasets.CIFAR100(root, train=True, download=True) self.assertEqual(len(dataset), 1) img, target = dataset[0] self.assertTrue(isinstance(img, PIL.Image.Image)) self.assertTrue(isinstance(target, int)) self.assertEqual(dataset.class_to_idx['fakedata'], target) dataset = torchvision.datasets.CIFAR100(root, train=False, download=True) self.assertEqual(len(dataset), 1) img, target = dataset[0] self.assertTrue(isinstance(img, PIL.Image.Image)) self.assertTrue(isinstance(target, int)) self.assertEqual(dataset.class_to_idx['fakedata'], target)