python, unittest: hay una manera de pasar opciones de línea de comandos a la aplicación

Question

Tengo un módulo que importa unittest y tiene algunos TestCases. Me gustaría para aceptar algunas opciones de línea de comando (por ejemplo a continuación, el nombre de un archivo de datos), pero cuando intento pasar la opción obtengo el mensaje "opción -i no reconocida". ¿Es posible tener opciones de prueba unittest + para la aplicación (nota: estoy usando optparse para manejar las opciones)? Gracias.

$ python test_app_data.py -i data_1.txt 

option -i not recognized 

=====================

seguimiento: se trata de una implementación de la solución sugerida:

import cfg_master #has the optparse option-handling code 

... 

if __name__ == '__main__':  
    #add you app's options here... 
    options_tpl = ('-i', '--in_dir', '-o', '--out_dir') 
    del_lst = [] 
    for i,option in enumerate(sys.argv): 
     if option in options_tpl: 
      del_lst.append(i) 
      del_lst.append(i+1) 

    del_lst.reverse() 
    for i in del_lst: 
     del sys.argv[i] 

    unittest.main() 

Answer 1

Sobre la base de la respuesta de Alex, en realidad es bastante fácil de hacer usando argparse:

if __name__ == '__main__': 
    parser = argparse.ArgumentParser() 
    parser.add_argument('--input', default='My Input') 
    parser.add_argument('filename', default='some_file.txt') 
    parser.add_argument('unittest_args', nargs='*') 

    args = parser.parse_args() 
    # TODO: Go do something with args.input and args.filename 

    # Now set the sys.argv to the unittest_args (leaving sys.argv[0] alone) 
    sys.argv[1:] = args.unittest_args 
    unittest.main() 

no he probado todas las banderas se puede pasar a unittest para ver si funcionan o no, pero al pasar la prueba nombres en funciona, por ejemplo:

python test.py --input=foo data.txt MyTest 

ejecuta MyTest con foo y data.txt.

Answer 2

En su sección if __name__ == '__main__':, que no nos está mostrando, que necesitará para optparse y luego del sys.argv[1:] antes de pasar el control a unittest código, por lo que este último código no trata de interpretar las opciones de línea de comandos de nuevo cuando tu Ya has tratado con ellos. (Es un poco más difícil tener algunas opciones propias y que también se transfieren a unittest, aunque se puede hacer si tiene necesidades tan complejas).

Answer 3

No debe tomar argumentos y opciones para ejecutar pruebas de unidad, ya que las hace ejecutar en condiciones diferentes y menos predecibles de esta manera. Debería averiguar por qué necesita ejecutar pruebas con diferentes datos y hacer que el conjunto de pruebas sea lo suficientemente completo como para cubrir el terreno de todos los conjuntos de datos sin que se ejecute de manera diferente cada vez.

Answer 4

Para pequeñas aplicaciones independientes, utilizo una opción centinela inicial (-t) y llamo unittest.main() antes de llamar argparse.ArgumentParser()

if __name__ == "__main__": 
    if len(sys.argv) > 1 and sys.argv[1] in ["-t", "--test"]: 
     del(sys.argv[1]) 
     sys.exit(unittest.main()) # pass sys.argv[ 

    p = argparse.ArgumentParser() 
    . . .