From b492c2a908b3c0a9ef709a072825b2cb790e2ec8 Mon Sep 17 00:00:00 2001 From: Finn Date: Wed, 25 Nov 2015 19:09:59 -0800 Subject: [PATCH] Update tests.py to allow a single file to be tested, to help work towards #22 --- tests.py | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/tests.py b/tests.py index ee06f4a..3e10342 100755 --- a/tests.py +++ b/tests.py @@ -39,13 +39,17 @@ def validate(path): print(" %sInvalid JSON!%s" % (RED, END)) return False -success = True -for directory, subdirs, files in os.walk('.'): - if len(files) > 0: - for f in files: - if f.endswith('.k'): - result = validate("%s/%s" % (directory, f)) - if not result: - success = False -if not success: - sys.exit(1) +if __name__ == "__main__": + success = True + if len(sys.argv) == 2: + success = validate(sys.argv[1]) + else: + for directory, subdirs, files in os.walk('.'): + if len(files) > 0: + for f in files: + if f.endswith('.k'): + result = validate("%s/%s" % (directory, f)) + if not result: + success = False + if not success: + sys.exit(1)