Python – Enter to raw_input automatically

python

Just as an example, it might seem illogical. I have a get_name function as below, and wanted to write a automated script to call this function and enter to the raw_input automatically.

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

The automated script as below, what command should I add to enter my value automatically?

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

Best Answer

You can also substitute stdin with StringIO (aka memory file) instead of real file. This way the entered text will be in your testing code instead of separate text file.

based on Anand S Kumar'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

Also, for more sophisticated testing of interactive commandline functions/tools you may want to check the pyexpect package.

Related Topic