diff options
| -rwxr-xr-x | suite/generator.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/suite/generator.py b/suite/generator.py index 27455914..8ef2280c 100755 --- a/suite/generator.py +++ b/suite/generator.py @@ -117,12 +117,17 @@ class TestBinaryNinjaAPI(unittest.TestCase): {1}{2} if __name__ == "__main__": + api_only = False if len(sys.argv) > 1: - if sys.argv[1] == '-v' or sys.argv[1] == '-V' or sys.argv[1] == '--verbose': - verbose = True + for i in range(1, len(sys.argv)): + if sys.argv[i] == '-v' or sys.argv[i] == '-V' or sys.argv[i] == '--verbose': + verbose = True + elif sys.argv[i] == '--api-only': + api_only = True test_suite = unittest.defaultTestLoader.loadTestsFromModule(api_test) - test_suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(TestBinaryNinjaAPI)) + if not api_only: + test_suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(TestBinaryNinjaAPI)) runner = unittest.TextTestRunner(verbosity=2) runner.run(test_suite) """ |