update unit test cases

Author: mingxiaoh

.travis.yml

@@ -42,7 +42,7 @@ script:
   - cd python
   - python setup.py install
   - cd ideep4py/tests
-  - CHAINER_TEST_GPU_LIMIT=0 pytest -m "not slow and not cudnn" ideep4py_tests
+  - ENALE_TRAVIS_TEST=1 pytest -s -v .
 
 
 addons:

python/ideep4py/tests/ideep4py_tests/test_convolution2d_py.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import unittest
 import numpy
@@ -24,11 +25,18 @@ def _set_cover_all(self, x, W):
                                  self.pw, d=self.dx))
 
 
+if bool(int(os.environ.get('ENALE_TRAVIS_TEST', '0'))):
+    bs_list = [1, 2, 4, 5, 8, 10, 16, 32, 64, ]
+else:
+    bs_list = [1, 2, 4, 5, 8, 10, 16, 32, 64, 96, 128, 192, 256, 512, ]
+print('bs_list: ', bs_list)
+
+
 @testing.parameterize(*testing.product({
     'dtype': [numpy.float32, ],
     'cover_all': [False, True],
     'channel': [1, 2, 4, 8, 10, ],
-    'bs': [1, 2, 4, 5, 8, 10, 16, 32, 64, 96, 128, 192, 256, 512, ],
+    'bs': bs_list,
     'with_bias': [True, ],
 }))
 @testing.fix_random()

python/ideep4py/tests/ideep4py_tests/test_pooling_2d_py.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import unittest
 
@@ -16,11 +17,17 @@
     print('*** testing directory is missing: %s' % ex)
     sys.exit(-1)
 
+if bool(int(os.environ.get('ENALE_TRAVIS_TEST', '0'))):
+    bs_list = [0, 1, 2, 4, 5, 6, 8, 10, 16, 24, 32, 64, ]
+else:
+    bs_list = [0, 1, 2, 4, 5, 6, 8, 10, 16, 24, 32, 64, 96, 128, 196, 256, ]
+print('bs_list: ', bs_list)
+
 
 @testing.parameterize(*testing.product({
     'dtype': [numpy.float32],
     'channel': [1, 2, 4, 8, 10, 16, 24, 32, 64, ],
-    'bs': [0, 1, 2, 4, 5, 6, 8, 10, 16, 24, 32, 64, 96, 128, 196, 256, ],
+    'bs': bs_list,
     'stride': [2, ],
 }))
 class TestPooling2DPyF32(unittest.TestCase):