Path: csiph.com!v102.xanadu-bbs.net!xanadu-bbs.net!feeder.erje.net!eu.feeder.erje.net!newsfeed.xs4all.nl!newsfeed4.news.xs4all.nl!xs4all!post.news.xs4all.nl!not-for-mail
To: python-list@python.org
From: Peter Otten <__peter__@web.de>
Subject: Re: Testing interactive code using raw_input
Date: Mon, 10 Mar 2014 17:57:19 +0100
Organization: None
References: <531de147$0$29994$c3e8da3$5496439d@news.astraweb.com>
Mime-Version: 1.0
Content-Type: text/plain; charset="ISO-8859-1"
Content-Transfer-Encoding: 7Bit
User-Agent: KNode/4.11.5
Precedence: list
Newsgroups: comp.lang.python
Message-ID: <mailman.8012.1394470659.18130.python-list@python.org>
Lines: 49
NNTP-Posting-Host: 2001:888:2000:d::a6
Xref: csiph.com comp.lang.python:68162

Steven D'Aprano wrote:

> Does anyone have any good hints for testing interactive code that uses
> raw_input, or input in Python 3?
> 
> A simple technique would be to factor out the interactive part, e.g. like
> this:
> 
> # Before
> def spam():
>     answer = raw_input(prompt)
>     return eggs(answer) + cheese(answer) + toast(answer)
> 
> # After
> def spam():
>     answer = raw_input(prompt)
>     return func(answer)
> 
> def func(s):
>     return eggs(s) + cheese(s) + toast(s)
> 
> 
> 
> and then test func. But how about times where it is inconvenient to
> factor out the raw_input stuff out of the function? E.g. suppose you have
> a function that takes some arguments, gathers some more values
> interactively, processes the lot, and then returns a result. With an
> automated test, I can provide the arguments, and check the result, but
> what are my options for *automatically* supplying input to raw_input?

https://pypi.python.org/pypi/mock

In Python 3 this is part of the standard library:

from unittest import mock

def sum_it():
    return "{} + {} = {}".format(input(), input(), input())

with mock.patch('builtins.input', side_effect="123"):
    assert sum_it() == "1 + 2 = 3"

I have not yet used it myself so far; instead I did something like

def sum_it(input=input):
   ...

and then passed a hand-crafted mock input function to increase coverage.