Wprowadź do raw_input automatycznie

Question

Przykładowo może wydawać się nielogiczne. Mam funkcję get_name, jak poniżej, i chciałem napisać automatyczny skrypt, aby zadzwonić do tej funkcji i wprowadzić automatycznie do raw_input.

def get_name(): 
    name = raw_input("Please enter your name : ") 
    print "Hi " + name 

Zautomatyzowany skrypt, jak poniżej, jakie polecenie należy dodać, aby automatycznie wprowadzić wartość?

def run(): 
    get_name() 
    // what should I add here? 

Answer 1

można przekierować stdin do pliku, a następnie raw_input() byłoby odczytać z tego pliku.

Przykład -

def run(): 
    import sys 
    f1 = sys.stdin 
    f = open('input.txt','r') 
    sys.stdin = f 
    get_name() 
    f.close() 
    sys.stdin = f1 

Należy pamiętać, że po wykonaniu - f = open('input.txt','r') i sys.stdin = f, raw_input() będzie czytać z pliku <filename>.

Po zakończeniu wywołania get_name() zamknij plik i przywróć standardowe wejście, używając sys.stdin = sys.__stdin__, jeśli chcesz przywrócić je do wejścia konsoli, możesz je przywrócić do wersji f1, która przywróci je do stwierdzić, że było to przed rozpoczęciem testu.

Należy pamiętać, że podczas przekierowywania wejść należy zachować ostrożność.

Answer 2

Do testowania możesz wywołać swój skrypt z wiersza poleceń z przekierowaniem IO - zobacz podproces w podręcznikach, ale dla szybkiego rozwiązania możesz zmienić swój kod w ten sposób, zauważ, że to nie testuje raw_input ale pozwala po prostu przetestować otaczająca kod:

def get_name (name=''): 
    """ Code to get the name for testing supply name int the call """ 
    if len(name) == 0: 
     name = raw_input("Please enter your name : ") 
    print "Hi " + name 

def run(): 
    get_name ("Fred") 

Answer 3

można również zastąpić stdin z StringIO (aka plik pamięci) zamiast rzeczywistej pliku. W ten sposób wpisany tekst znajdzie się w kodzie testowym zamiast w osobnym pliku tekstowym.

podstawie Anand Kumar S'S (+1):

def run(): 
    import sys 
    import StringIO 
    f1 = sys.stdin 
    f = StringIO.StringIO('entered text') # <-- HERE 
    sys.stdin = f 
    get_name() 
    f.close() 
    sys.stdin = f1 

Ponadto, dla bardziej zaawansowanych testów interaktywnych funkcji wiersza poleceń/narzędzia może chcesz sprawdzić pakiet pyexpect.

Answer 4

Inną opcją jest, aby funkcja wejścia parametr, zalegających do raw_input:

def get_name(infunc=raw_input): 
    name = infunc("Please enter your name : ") 
    print "Hi " + name 

Następnie do celów testowych można przekazać w funkcji, która robi co trzeba:

get_name(lambda prompt: "John Smith")