Fix tests

Author: rickwierenga

tensorflow_datasets/image/imagewang_test.py

@@ -1,4 +1,4 @@
-""" Tests for Imagewang contains Imagenette and Imagewoof combined. """
+""" Tests for Imagewang """
 
 from __future__ import absolute_import
 from __future__ import division
@@ -12,26 +12,26 @@ class ImagewangFullSizeTest(testing.DatasetBuilderTestCase):
   DATASET_CLASS = imagewang.Imagewang
   BUILDER_CONFIG_NAMES_TO_TEST = ['full-size']
   SPLITS = {
-      "train": 1,
-      "validation": 1,
+      "train": 4,
+      "validation": 4,
   }
 
 
 class Imagewang320Test(testing.DatasetBuilderTestCase):
   DATASET_CLASS = imagewang.Imagewang
   BUILDER_CONFIG_NAMES_TO_TEST = ['320px']
   SPLITS = {
-      "train": 1,
-      "validation": 1,
+      "train": 4,
+      "validation": 4,
   }
 
 
 class Imagewang160Test(testing.DatasetBuilderTestCase):
   DATASET_CLASS = imagewang.Imagewang
   BUILDER_CONFIG_NAMES_TO_TEST = ['160px']
   SPLITS = {
-      "train": 1,
-      "validation": 1,
+      "train": 4,
+      "validation": 4,
   }
 
 

tensorflow_datasets/testing/imagewang.py

@@ -36,7 +36,7 @@
                     "Path to tensorflow_datasets directory")
 FLAGS = flags.FLAGS
 
-_IMAGE_FILENAME = 'test_image'
+_IMAGE_FILENAME = 'test_image.JPEG'
 _LABEL_DIRNAME = 'label'
 
 _SIZES = ["full-size", "320px", "160px"]