added tests for remaining SVHN splits

Author: Philip Meier

test/test_datasets.py

@@ -140,6 +140,12 @@ def test_svhn(self, mock_check):
             dataset = torchvision.datasets.SVHN(root, split="train")
             self.generic_classification_dataset_test(dataset, num_images=2)
 
+            dataset = torchvision.datasets.SVHN(root, split="test")
+            self.generic_classification_dataset_test(dataset, num_images=2)
+
+            dataset = torchvision.datasets.SVHN(root, split="extra")
+            self.generic_classification_dataset_test(dataset, num_images=2)
+
 
 if __name__ == '__main__':
     unittest.main()