Add coverage to onnx tests.

apache · Dec 16, 2020 · 86270bb · 86270bb
1 parent 0d19537
commit 86270bb
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh
@@ -1273,16 +1273,17 @@ unittest_centos7_gpu() {
 }
 
 integrationtest_ubuntu_cpu_onnx() {
-	set -ex
-	export PYTHONPATH=./python/
-	export MXNET_SUBGRAPH_VERBOSE=0
-	export DMLC_LOG_STACK_TRACE_DEPTH=10
-	tests/python-pytest/onnx/backend_test.py
-	pytest tests/python-pytest/onnx/mxnet_export_test.py
-	pytest tests/python-pytest/onnx/test_models.py
-	pytest tests/python-pytest/onnx/test_node.py
-	pytest tests/python-pytest/onnx/test_operators.py
-	pytest tests/python-pytest/onnx/test_onnxruntime.py
+    set -ex
+    export PYTHONPATH=./python/
+    export MXNET_SUBGRAPH_VERBOSE=0
+    export DMLC_LOG_STACK_TRACE_DEPTH=10
+    tests/python-pytest/onnx/backend_test.py
+    COV_ARG="--cov --cov-report xml:tests_onnx.xml --cov-append"
+    pytest $COV_ARG -v tests/python-pytest/onnx/mxnet_export_test.py
+    pytest $COV_ARG -v tests/python-pytest/onnx/test_models.py
+    pytest $COV_ARG -v tests/python-pytest/onnx/test_node.py
+    pytest $COV_ARG -v tests/python-pytest/onnx/test_operators.py
+    pytest $COV_ARG -v tests/python-pytest/onnx/test_onnxruntime.py
 }
 
 integrationtest_ubuntu_gpu_python() {