Python - argparse

#!/usr/bin/env python

'''
$ python run.py -t pierwszy -p iOS
$ python run.py --test pierwszy --platform iOS
'''

'''
usage: run.py [-h] [-t TEST] [-p PLATFORM]
               [unittest_args [unittest_args ...]]

Pass test and platform

positional arguments:
  unittest_args

optional arguments:
  -h, --help            show this help message and exit
  -t TEST, --test TEST  test name (default: all)
  -p PLATFORM, --platform PLATFORM
                        platform name (default: Android_7)
'''

import argparse

def runner():

    parser = argparse.ArgumentParser(description='Pass test and platform')

    parser.add_argument('-t', '--test', default='all', help='test name (default: all)')
    parser.add_argument('-p', '--platform', default='Android_7', help='platform name (default: Android_7)')
    parser.add_argument('unittest_args', nargs='*')

    args = parser.parse_args()

    #print('Platform:', args.platform)
    #print('Test:', args.test)

    loader = unittest.TestLoader()

    if args.test == "all":
        names = loader.discover(start_dir="./tests", pattern="test*.py", top_level_dir=None)
        unittest.TextTestRunner(verbosity=2).run(names)
    elif len(var1) > 1:
        names = loader.discover(start_dir="./tests", pattern=args.test + ".py")
        unittest.TextTestRunner(verbosity=2).run(names)
    else:
        parser.error("wrong test name")

    print("test passed into run.py =", args.test)
    print("platform passed into run.py =", args.platform)

    return args.platform


runner()