From 5a34805d05420c0866d14ffccd700e44a66cfa62 Mon Sep 17 00:00:00 2001 From: Behdad Esfahbod Date: Fri, 9 Nov 2018 11:54:40 -0500 Subject: [PATCH] [varLib.featureVars] Improve test --- Tests/varLib/featureVars_test.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/Tests/varLib/featureVars_test.py b/Tests/varLib/featureVars_test.py index e2bda323b..016f0c500 100644 --- a/Tests/varLib/featureVars_test.py +++ b/Tests/varLib/featureVars_test.py @@ -20,12 +20,19 @@ def test_explosion(n = 10): if __name__ == "__main__": import sys from pprint import pprint + quiet = False args = {} if len(sys.argv) > 1: + if sys.argv[1] == '-q': + quiet = True + del sys.argv[1] args['n'] = int(sys.argv[1]) input, output = test_explosion(**args) - print("Input:") - pprint(input) - print() - print("Output:") - pprint(output) + if quiet: + print(len(output)) + else: + print("Input:") + pprint(input) + print() + print("Output:") + pprint(output)