[Test Mv]mv dataset test

python/CMakeLists.txt

@@ -149,7 +149,6 @@ add_custom_target(paddle_python ALL
 set(PADDLE_PYTHON_PACKAGE_DIR ${CMAKE_CURRENT_BINARY_DIR}/dist/)
 
 if(WITH_TESTING)
-  add_subdirectory(paddle/dataset/tests)
   add_subdirectory(paddle/fluid/tests)
   add_subdirectory(paddle/fluid/contrib/tests)
   add_subdirectory(paddle/static/quantization/tests)

test/CMakeLists.txt

@@ -17,7 +17,7 @@ if(WITH_TESTING)
   add_subdirectory(custom_kernel)
   # add_subdirectory(custom_op)
   # add_subdirectory(custom_runtime)
-  # add_subdirectory(dataset)
+  add_subdirectory(dataset)
   # add_subdirectory(distributed_passes)
   # add_subdirectory(distribution)
   # add_subdirectory(dygraph_to_static)

test/dataset/CMakeLists.txt

@@ -0,0 +1,9 @@
+file(
+  GLOB TEST_OPS
+  RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
+  "test_*.py")
+string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
+
+foreach(src ${TEST_OPS})
+  py_test(${src} SRCS ${src}.py)
+endforeach()

python/paddle/dataset/tests/test_image.py → test/dataset/test_image.py

@@ -17,8 +17,7 @@
 Description:
     This scipt test image resize,flip and chw.
 """
-
-import sys
+import os
 import unittest
 
 import numpy as np
@@ -35,7 +34,9 @@ class Image(unittest.TestCase):
 
     def test_resize_flip_chw(self):
         """resize"""
-        imgdir = sys.argv[0].replace('test_image.py', 'cat.jpg')
+        imgdir = os.path.join(
+            os.path.dirname(os.path.abspath(__file__)), 'cat.jpg'
+        )
         images = image.load_image(imgdir)
         images = image.resize_short(images, 256)
         self.assertEqual(256, min(images.shape[:2]))