python, unittest: czy istnieje sposób przekazywania opcji wiersza poleceń do aplikacji

Question

Mam moduł, który importuje unittest i ma kilka testcases. Chciałbym zaakceptować niektóre opcje wiersza poleceń (na przykład poniżej, nazwa pliku danych), , ale kiedy spróbuję przekazać opcję, otrzymuję komunikat "opcja -i nie rozpoznana". Czy jest możliwe, aby unittest + zapewnić opcje do aplikacji (uwaga: używam opcji optparse do obsługi opcji)? Dzięki.

$ python test_app_data.py -i data_1.txt 

option -i not recognized 

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

follow-up: to jest realizacja proponowanego rozwiązania:

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

Opierając się na odpowiedź Alexa, to faktycznie dość łatwe do zrobienia przy użyciu 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() 

nie testowałem wszystkie flagi można przekazać do unittest, aby zobaczyć, czy pracują, czy nie, ale przejściu testu nazwy w działa, np:

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

działa MyTest z foo i data.txt.

Answer 2

W sekcji if __name__ == '__main__':, których nie jesteś pokazując nam, musisz optparse a następnie del sys.argv[1:] zanim przejdzie kontrolę do kodu unittest, tak że ten ostatni kod nie próbować interpretować opcji wiersza poleceń ponownie kiedy yo już z nimi poradziłeś. (Trochę trudniej jest mieć niektóre opcje własnego i również przekazać trochę do unittest, choć można to zrobić, jeśli masz tak złożone potrzeby).

Answer 3

Nie powinieneś brać argumentów i opcji, aby uruchamiać unittests, ponieważ uruchamiasz je w różnych, mniej przewidywalnych warunkach w ten sposób. Powinieneś dowiedzieć się, dlaczego musisz uruchamiać testy z różnymi danymi i sprawić, że zestaw testów będzie na tyle kompletny, aby pokryć cały zbiór danych, nie za każdym razem uruchamiając go inaczej.

Answer 4

Dla małych aplikacjach autonomicznych, używam początkową opcję wartowniczy (-t) i wywołać unittest.main() przed wywołaniem 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() 
    . . .