Python测试原始输入，if语句中的原始输入

Question

我目前正在测试我的 python 代码并且对 raw_input 有疑问。这是我的功能:

def answer():
    ans = raw_input('enter yes or no')
    if ans == 'yes':
        print 'you entered yes'
        return 'yes'
    if ans == 'no':
        some_value = raw_input('enter some value: ')
        print 'you entered no'
        return some_value

我正在以这种方式测试第一个 if 语句:

with mock.patch('__builtin__.raw_input', return_value= 'yes'):
    assert answer() == 'yes'

但是我该如何检查 no 语句呢？我如何在模拟中进行模拟？

Answer 1

使用 side_effect :

with mock.patch('__builtin__.raw_input', side_effect=['yes']):
    assert answer() == 'yes'
with mock.patch('__builtin__.raw_input', side_effect=['no', 'maybe']):
    assert answer() == 'maybe'

根据 mock documentation :

If side_effect is an iterable then each call to the mock will return the next value from the iterable. The side_effect can also be any iterable object. Repeated calls to the mock will return values from the iterable (until the iterable is exhausted and a StopIteration is raised):

>>>
>>> m = MagicMock(side_effect=[1, 2, 3])
>>> m()
1
>>> m()
2
>>> m()
3
>>> m()
Traceback (most recent call last):
  ...
StopIteration