Collecting coiverage when running python tests

kubeflow · Mar 1, 2019 · 04ee5c6 · 04ee5c6
1 parent 18878f1
commit 04ee5c6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -54,9 +54,10 @@ matrix:
         - cd $TRAVIS_BUILD_DIR/sdk/python
         - ./build.sh /tmp/kfp.tar.gz
         - pip install /tmp/kfp.tar.gz
-        - python tests/dsl/main.py
-        - python tests/compiler/main.py
-        - python -m unittest discover --verbose --start-dir tests --top-level-directory=..
+        - pip install coverage
+        - coverage run --include=*/kfp/* --append python tests/dsl/main.py
+        - coverage run --include=*/kfp/* --append python tests/compiler/main.py
+        - coverage run --include=*/kfp/* --append -m unittest discover --verbose --start-dir tests --top-level-directory=..
 
         # Component SDK tests
         - cd $TRAVIS_BUILD_DIR/component_sdk/python