Supports excluding tests.

Author: springcomp

bin/jp-compliance

@@ -73,11 +73,14 @@ _bname = os.path.basename
 class ComplianceTestRunner(object):
     TEST_DIR = _pjoin(_dname(_dname(_abs(__file__))), 'tests')
 
-    def __init__(self, exe=None, tests=None, test_dir=None):
+    def __init__(self, exe=None, tests=None, test_dir=None, exclude=None):
         if test_dir is None:
             test_dir = self.TEST_DIR
         self.test_dir = test_dir
         self.tests = tests
+        self.excludes = []
+        if exclude != None:
+            self.excludes = exclude.split(',')
         self.jp_executable = exe
         self.had_failures = False
 
@@ -227,8 +230,9 @@ class ComplianceTestRunner(object):
         for root, _, filenames in os.walk(self.test_dir):
             for filename in filenames:
                 if filename.endswith('.json'):
-                    full_path = _pjoin(root, filename)
-                    yield full_path
+                    if not(filename in [item + '.json' for item in self.excludes]):
+                        full_path = _pjoin(root, filename)
+                        yield full_path
 
 
 def display_available_tests(test_files):
@@ -266,8 +270,10 @@ def main():
                               'specified, no tests will actually be run.'))
     parser.add_argument('-s', '--stop-first-fail', action='store_true',
                         help='Stop running tests after a single test fails.')
+    parser.add_argument('-x', '--exclude',
+                        help='Exclude tests from running.')
     args = parser.parse_args()
-    runner = ComplianceTestRunner(args.exe, args.tests, args.test_dir)
+    runner = ComplianceTestRunner(args.exe, args.tests, args.test_dir, args.exclude)
     if args.list:
         display_available_tests(runner.get_compliance_test_files())
     else: