Index: catapult_build/run_with_typ.py |
diff --git a/catapult_build/run_with_typ.py b/catapult_build/run_with_typ.py |
index b9c69f2e88db8e8169d48a419a76c5d2f48295b3..b90398edfa1bf08feb0f8ca4f17ce3280ba16ae2 100644 |
--- a/catapult_build/run_with_typ.py |
+++ b/catapult_build/run_with_typ.py |
@@ -9,7 +9,7 @@ import os |
import sys |
-def Run(top_level_dir, path=None): |
+def Run(top_level_dir, path=None, suffixes=None): |
"""Runs a set of Python tests using typ. |
Args: |
@@ -19,6 +19,8 @@ def Run(top_level_dir, path=None): |
Returns: |
An exit code (0 for success, otherwise non-zero). |
""" |
+ if not suffixes: |
+ suffixes = ['*_test.py', '*_unittest.py'] |
typ_path = os.path.abspath(os.path.join( |
os.path.dirname(__file__), os.path.pardir, 'third_party', 'typ')) |
_AddToPathIfNeeded(typ_path) |
@@ -26,7 +28,8 @@ def Run(top_level_dir, path=None): |
return typ.main( |
top_level_dir=top_level_dir, |
path=(path or []), |
- coverage_source=[top_level_dir]) |
+ coverage_source=[top_level_dir], |
+ suffixes=suffixes) |
def _AddToPathIfNeeded(path): |