<p>简而言之,答案是<a href="http://en.wikipedia.org/wiki/Monkey_patch" rel="nofollow noreferrer">monkey patch</a><code>raw_input()</code>。</p>
<p>在<a href="https://stackoverflow.com/questions/5812333/how-to-display-the-redirected-stdin-in-python">How to display the redirected stdin in Python?</a>的答案中有一些很好的例子</p>
<p>下面是一个简单的例子,使用一个<code>lambda</code>来丢弃提示并返回我们想要的内容。</p>
<h2>测试中的系统</h2>
<pre><code>cat ./name_getter.py
#!/usr/bin/env python
class NameGetter(object):
def get_name(self):
self.name = raw_input('What is your name? ')
def greet(self):
print 'Hello, ', self.name, '!'
def run(self):
self.get_name()
self.greet()
if __name__ == '__main__':
ng = NameGetter()
ng.run()
$ echo Derek | ./name_getter.py
What is your name? Hello, Derek !
</code></pre>
<h2>测试用例:</h2>
<pre><code>$ cat ./t_name_getter.py
#!/usr/bin/env python
import unittest
import name_getter
class TestNameGetter(unittest.TestCase):
def test_get_alice(self):
name_getter.raw_input = lambda _: 'Alice'
ng = name_getter.NameGetter()
ng.get_name()
self.assertEquals(ng.name, 'Alice')
def test_get_bob(self):
name_getter.raw_input = lambda _: 'Bob'
ng = name_getter.NameGetter()
ng.get_name()
self.assertEquals(ng.name, 'Bob')
if __name__ == '__main__':
unittest.main()
$ ./t_name_getter.py -v
test_get_alice (__main__.TestNameGetter) ... ok
test_get_bob (__main__.TestNameGetter) ... ok
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
</code></pre>