import sys
from doctest import _SpoofOut

def foo():
    input("prompt:")
    print()

original_stdout = sys.stdout

try:
    sys.stdout = _SpoofOut()
    foo()
finally:
    sys.stdout = original_stdout