OpenAttack

1
import argparse
2

3

4
def main(args,):
5
    import unittest
6
    import os, sys
7

8
    TEST_BASE = os.path.dirname(os.path.abspath(__file__))
9
    test_cases = unittest.defaultTestLoader.discover(TEST_BASE, pattern=args.file)
10
    runner = unittest.runner.TextTestRunner(verbosity=args.verbosity)
11
    ret = runner.run(test_cases)
12
    sys.exit(len(ret.failures) + len(ret.errors))
13

14

15
if __name__ == "__main__":
16
    parser = argparse.ArgumentParser()
17
    parser.add_argument(
18
        "--file", default="test_*.py", type=str,
19
    )
20
    parser.add_argument(
21
        "--verbosity", default=2, type=int,
22
    )
23
    args = parser.parse_args()
24
    main(args)
25