import unittest, argparse, gradescope_utils
from gradescope_utils.autograder_utils.json_test_runner import JSONTestRunner

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='''run unit tests for ECE 401 MP.''')
    parser.add_argument('-j','--json',action='store_true',help='''Results in JSON format.''')
    args = parser.parse_args()
    suite = unittest.defaultTestLoader.discover('tests')
    if args.json:
        JSONTestRunner(visibility='visible').run(suite)
    else:
        result = unittest.TextTestRunner().run(suite)
