Python знущається з кількох повернених значень

Я використовую pythons mock.patch і хотів би змінити значення повернення для кожного виклику. Ось застереження: функція, яку виправляють, не має входів, тому я не можу змінити повернене значення на основі вхідних даних.

Ось мій код для довідок.

def get_boolean_response():
    response = io.prompt('y/n').lower()
    while response not in ('y', 'n', 'yes', 'no'):
        io.echo('Not a valid input. Try again'])
        response = io.prompt('y/n').lower()

    return response in ('y', 'yes')

Мій тестовий код:

@mock.patch('io')
def test_get_boolean_response(self, mock_io):
    #setup
    mock_io.prompt.return_value = ['x','y']
    result = operations.get_boolean_response()

    #test
    self.assertTrue(result)
    self.assertEqual(mock_io.prompt.call_count, 2)

io.promptце просто незалежна від платформи (python 2 і 3) версія "введення". Тому в кінцевому рахунку я намагаюся знущатися над введенням користувачів. Я намагався використовувати список для повернення значення, але це не працює.

Видно, що якщо значення повернення щось недійсне, я просто отримаю тут нескінченний цикл. Тому мені потрібен спосіб врешті-решт змінити значення повернення, щоб мій тест фактично закінчився.

(Ще одним можливим способом відповісти на це питання може бути пояснення, як я міг імітувати введення користувача в тестовому модулі)

Не є дублем цього питання головним чином тому, що я не маю можливості змінювати дані.

Один із зауважень відповіді на це запитання - це те саме, але жодної відповіді / коментаря не надано.

Ви можете призначити Iterable To side_effect, і знущатися поверне таке значення в послідовності кожен раз , коли вона називається:

>>> from unittest.mock import Mock
>>> m = Mock()
>>> m.side_effect = ['foo', 'bar', 'baz']
>>> m()
'foo'
>>> m()
'bar'
>>> m()
'baz'

Цитування Mock()документації :

Якщо side_effect є ітерабельним, то кожен виклик макету повертає наступне значення з ітерабельного.

Як осторонь, тест неresponse is not 'y' or 'n' or 'yes' or 'no' буде працювати; ви запитуєте, чи вираз є істинним, чи істинним (завжди так, не порожній рядок завжди є істинним) тощо. Різні вирази з будь-якої сторони операторів незалежні . Див. Як перевірити одну змінну на кілька значень?(response is not 'y')'y'or

Ви також не повинні використовувати isдля тестування на рядок. Інтерпретатор CPython може використовувати повторно рядкові об'єкти за певних обставин , але це не поведінка, на яку слід розраховувати.

Використовуйте:

response not in ('y', 'n', 'yes', 'no')

натомість; це використовуватиме тести рівності ( ==), щоб визначити, чи responseпосилається рядок з однаковим вмістом (значення).

Це ж стосується і response == 'y' or 'yes'; використовувати response in ('y', 'yes')замість цього.