Skip not yet supported onnx backend test cases in non CI environment (#…

…229)
onnx · Feb 13, 2018 · f29968c · f29968c
1 parent d530daa
commit f29968c
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tests/onnx_backend_test.py b/tests/onnx_backend_test.py
@@ -14,14 +14,15 @@
 pytest_plugins = 'onnx.backend.test.report',
 
 backend_test = onnx.backend.test.BackendTest(c2, __name__)
-# Skip vgg to speed up CI, skip others because onnx-caffe2 does not support them yet.
-ci_blacklist = (r'(test_vgg19|test_vgg'  # Speed up CI.
-                '|test_ceil.*|test_floor.*'  # Does not support Ceil and Floor.
-                '|test_hardsigmoid.*|test_pow.*'  # Does not support Hardsigmoid and Pow.
-                '|test_mean.*|test_hardmax.*'  # Does not support Mean and Hardmax.
-                ')')
+
+backend_test.exclude(r'(test_ceil|test_floor'  # Does not support Ceil and Floor.
+                     '|test_hardsigmoid|test_pow'  # Does not support Hardsigmoid and Pow.
+                     '|test_mean|test_hardmax)')  # Does not support Mean and Hardmax.
+
+# Skip vgg to speed up CI
 if 'CI' in os.environ:
-    backend_test.exclude(ci_blacklist)
+    backend_test.exclude(r'(test_vgg19|test_vgg)')
+
 # import all test cases at global scope to make them visible to python.unittest
 globals().update(backend_test
                  .enable_report()