Skip to content

Commit 997cfb1

Browse files
remove old CIFAR tests and fake data generation (#3447)
Reviewed By: fmassa Differential Revision: D26756258 fbshipit-source-id: eaf36956b2f341276088ed7702a484a9b93d448e
1 parent bd06036 commit 997cfb1

File tree

1 file changed

+1
-33
lines changed

1 file changed

+1
-33
lines changed

test/test_datasets.py

Lines changed: 1 addition & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import torchvision
1111
from torchvision.datasets import utils
1212
from common_utils import get_tmp_dir
13-
from fakedata_generation import mnist_root, cifar_root, imagenet_root, \
13+
from fakedata_generation import mnist_root, imagenet_root, \
1414
cityscapes_root, svhn_root, places365_root, widerface_root, stl10_root
1515
import xml.etree.ElementTree as ET
1616
from urllib.request import Request, urlopen
@@ -173,38 +173,6 @@ def test_widerface(self, mock_check_integrity):
173173
img, target = dataset[0]
174174
self.assertTrue(isinstance(img, PIL.Image.Image))
175175

176-
@mock.patch('torchvision.datasets.cifar.check_integrity')
177-
@mock.patch('torchvision.datasets.cifar.CIFAR10._check_integrity')
178-
def test_cifar10(self, mock_ext_check, mock_int_check):
179-
mock_ext_check.return_value = True
180-
mock_int_check.return_value = True
181-
with cifar_root('CIFAR10') as root:
182-
dataset = torchvision.datasets.CIFAR10(root, train=True, download=True)
183-
self.generic_classification_dataset_test(dataset, num_images=5)
184-
img, target = dataset[0]
185-
self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target)
186-
187-
dataset = torchvision.datasets.CIFAR10(root, train=False, download=True)
188-
self.generic_classification_dataset_test(dataset)
189-
img, target = dataset[0]
190-
self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target)
191-
192-
@mock.patch('torchvision.datasets.cifar.check_integrity')
193-
@mock.patch('torchvision.datasets.cifar.CIFAR10._check_integrity')
194-
def test_cifar100(self, mock_ext_check, mock_int_check):
195-
mock_ext_check.return_value = True
196-
mock_int_check.return_value = True
197-
with cifar_root('CIFAR100') as root:
198-
dataset = torchvision.datasets.CIFAR100(root, train=True, download=True)
199-
self.generic_classification_dataset_test(dataset)
200-
img, target = dataset[0]
201-
self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target)
202-
203-
dataset = torchvision.datasets.CIFAR100(root, train=False, download=True)
204-
self.generic_classification_dataset_test(dataset)
205-
img, target = dataset[0]
206-
self.assertEqual(dataset.class_to_idx[dataset.classes[0]], target)
207-
208176
@unittest.skipIf('win' in sys.platform, 'temporarily disabled on Windows')
209177
def test_cityscapes(self):
210178
with cityscapes_root() as root:

0 commit comments

Comments
 (0)