For test

paddle/scripts/paddle_build.sh

@@ -315,7 +315,8 @@ function run_test() {
 EOF
         echo $http_proxy
         echo $https_proxy
-        ctest -V
+        ctest -R test_parallel_executor_fetch_feed -V
+        ctest -R test_dist_se_resnext -V
         # make install should also be test when unittest
         make install -j `nproc`
         pip install /usr/local/opt/paddle/share/wheels/*.whl

python/paddle/dataset/flowers.py

@@ -41,6 +41,7 @@ from paddle.reader import *
 import os
 import numpy as np
 from multiprocessing import cpu_count
+import six
 from six.moves import cPickle as pickle
 from six.moves import zip
 __all__ = ['train', 'test', 'valid']

python/paddle/fluid/tests/unittests/CMakeLists.txt

@@ -64,7 +64,7 @@ if(WITH_DISTRIBUTE)
 endif()
 py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf SERIAL)
 py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed SERIAL)
-set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 600)
+set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 300)
 py_test_modules(test_dist_transformer MODULES test_dist_transformer SERIAL)
 py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext SERIAL)
 py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL)