Activates coverage for Trax in OSS.

PiperOrigin-RevId: 308896581

Author: afrozenator

.coveragerc

@@ -0,0 +1,16 @@
+[run]
+source =
+    trax/
+omit =
+    *_test.py
+    */site-packages/*
+
+[report]
+omit =
+    */site-packages/*
+exclude_lines =
+    pragma: no cover
+    def __repr__
+    raise NotImplementedError
+    if __name__ == .__main__.:
+

oss_scripts/oss_tests.sh

@@ -16,6 +16,9 @@
 
 set -v  # print commands as they're executed
 
+# aliases aren't expanded in non-interactive shells by default.
+shopt -s expand_aliases
+
 # Instead of exiting on any failure with "set -e", we'll call set_status after
 # each command and exit $STATUS at the end.
 STATUS=0
@@ -42,6 +45,9 @@ fi
 python -c "import trax"
 set_status
 
+# Run pytest with coverage.
+alias pytest='coverage run -m pytest'
+
 # Check tests, separate out directories for easy triage.
 
 # Disabled test fails with "disable-warnings", tested separately.
@@ -101,4 +107,7 @@ jupyter nbconvert --ExecutePreprocessor.kernel_name=python3 \
   trax/intro.ipynb;
 set_status
 
+# Print coverage report.
+coverage report -m
+
 exit $STATUS