Added EMNIST Test

us · us · commit 976bb826994b · 2019-03-09T05:54:53.000+03:00
diff --git a/tensorflow_datasets/image/mnist_test.py b/tensorflow_datasets/image/mnist_test.py
@@ -50,5 +50,10 @@ class KMNISTTest(MNISTTest):
     DATASET_CLASS = mnist.KMNIST
 
 
+class EMNISTTest(MNISTTest):
+    DATASET_CLASS = mnist.EMNIST
+    BUILDER_CONFIG_NAMES_TO_TEST = ["test"]
+
+
 if __name__ == "__main__":
   testing.test_main()
diff --git a/tensorflow_datasets/testing/mnist.py b/tensorflow_datasets/testing/mnist.py
@@ -71,7 +71,7 @@ def write_label_file(filename, num_labels):
 
 
 def main(_):
-  for mnist in ["mnist", "fashion_mnist", "kmnist"]:
+  for mnist in ["mnist", "fashion_mnist", "kmnist", "emnist"]:
     output_dir = mnist_dir(mnist)
     test_utils.remake_dir(output_dir)
     write_image_file(os.path.join(output_dir, _TRAIN_DATA_FILENAME), 10)
diff --git a/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-test-images-idx3-ubyte b/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-test-images-idx3-ubyte
diff --git a/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-test-labels-idx1-ubyte b/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-test-labels-idx1-ubyte
@@ -0,0 +1 @@
+11111111
diff --git a/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-train-images-idx3-ubyte b/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-train-images-idx3-ubyte
diff --git a/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-train-labels-idx1-ubyte b/tensorflow_datasets/testing/test_data/fake_examples/emnist/emnist-test-train-labels-idx1-ubyte
@@ -0,0 +1 @@
+11111111
