tests/utils/python/testrunner.py: add optional pattern argument

[babeltrace.git] / tests / utils / python / testrunner.py

diff --git a/tests/utils/python/testrunner.py b/tests/utils/python/testrunner.py
index ef1872fa36b3703914d00c2ed5218adda3124cd5..175c8fd188c8ba2b43dde2412fe21f7c3ae8712d 100644 (file)
--- a/tests/utils/python/testrunner.py
+++ b/tests/utils/python/testrunner.py
@@ -27,7 +27,13 @@ import sys
 
 if __name__ == '__main__':
     loader = unittest.TestLoader()
-    tests = loader.discover(sys.argv[1])
+
+    if len(sys.argv) >= 3:
+        pattern = sys.argv[2]
+    else:
+        pattern = 'test*.py'
+
+    tests = loader.discover(sys.argv[1], pattern)
     runner = TAPTestRunner()
     runner.set_stream(True)
     runner.set_format('{method_name}')