a decorator to run a module and capture stdin/stdout

run_module_dec.py

def run(module_path):
    # run a module
    # capture standard input and output
    def wrapper(input):
        stream_in = StringIO(input)
        stream_out = StringIO()
        stdin_ = sys.stdin
        stdout_ = sys.stdout
        sys.stdin = stream_in
        sys.stdout = stream_out

        runpy.run_path(module_path)

        sys.stdin = stdin_
        sys.stdout = stdout_
        return stream_out.getvalue()
    wrapper.__name__ = module_path
    return wrapper