python, unittest : 명령 행 옵션을 앱에 전달하는 방법이 있나요

Question

unittest를 가져오고 일부 TestCases가있는 모듈이 있습니다. 일부 명령 행 옵션 (예 : 아래의 데이터 파일 이름)을 허용하려면, 하지만 옵션을 전달하려고 할 때 "옵션 -i not recognized"메시지가 나타납니다. unittest + 앱에 옵션을 제공 할 수 있습니까 (참고 : 옵션을 처리하기 위해 optparse를 사용하고 있습니다)? 감사.

$ python test_app_data.py -i data_1.txt 

option -i not recognized 

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

후속은 :

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() 

:이 제안 된 솔루션의 구현입니다 당신이 unittest 코드에 제어를 전달하기 전에 후자의 코드가 명령 행 옵션을 해석하려고하지 않도록 당신이 우리를 표시하지 않는 당신의 if __name__ == '__main__': 섹션에서

Answer 1

건물 알렉스의 대답에, 그것은 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() 

난 당신이 그들이 작동 여부 있는지 확인하기 위해 유닛 테스트에 통과 할 수 플래그의 모든 테스트를하지 않은,하지만 시험을 통과 수행 작업, 예에서 이름 : foo와 data.txt와 MYTEST

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

실행됩니다.

작은 독립형 앱의

Answer 2

, 당신은 다음 del sys.argv[1:]optparse에 필요합니다 다시 때 요 이미 그것들을 다루었습니다. 약간의 옵션을 및으로 지정하는 것이 조금 더 어려워서 unittest으로 전달할 수 있지만 복잡한 요구가있는 경우 수행 할 수 있습니다.

Answer 3

unittests를 실행하기 위해 인수와 옵션을 취하지 않아야합니다. 단위 테스트를 다른 방법으로 예측하기 어려운 상태로 실행하게하십시오. 왜 다른 데이터로 테스트를 실행해야하는지 파악해야하고, 매번 다르게 실행하지 않고 모든 데이터 세트의 기반을 충분히 커버 할 수있는 테스트 슈트를 만들 수 있어야합니다.

Answer 4

, 나는 초기 감시 옵션 (-t)를 사용하고 예, 일반적인 용어() 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() 
    . . .