Python simula vários valores de retorno

Estou usando pythons mock.patch e gostaria de alterar o valor de retorno para cada chamada. Aqui está a ressalva: a função que está sendo corrigida não possui entradas, portanto, não posso alterar o valor de retorno com base na entrada.

Aqui está o meu código para referência.

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')

Meu código de teste:

@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é apenas uma versão independente da plataforma (python 2 e 3) de "input". Então, finalmente, estou tentando zombar da entrada dos usuários. Eu tentei usar uma lista para o valor de retorno, mas isso não parece funcionar.

Você pode ver que, se o valor de retorno for algo inválido, receberei apenas um loop infinito aqui. Então, preciso de uma maneira de alterar o valor de retorno, para que meu teste realmente termine.

(outra maneira possível de responder a essa pergunta poderia ser explicar como eu poderia imitar a entrada do usuário em um teste de unidade)

Não estou enganado com esta questão principalmente porque não tenho a capacidade de variar as entradas.

Um dos comentários da resposta sobre esta pergunta está na mesma linha, mas nenhuma resposta / comentário foi fornecido.

Você pode atribuir um iterável a side_effect, e a simulação retornará o próximo valor na sequência sempre que for chamado:

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

Citando a Mock()documentação :

Se side_effect for iterável, cada chamada para a simulação retornará o próximo valor da iterável.

Como um aparte, o teste response is not 'y' or 'n' or 'yes' or 'no'vai não trabalhar; você está perguntando se a expressão (response is not 'y')é verdadeira ou 'y'verdadeira (sempre é o caso, uma seqüência de caracteres não vazia é sempre verdadeira) etc. As várias expressões de ambos os lados dos oroperadores são independentes . Consulte Como faço para testar uma variável em relação a vários valores?

Você também não deve usar ispara testar uma string. O intérprete CPython pode reutilizar objetos de string em determinadas circunstâncias , mas não é com esse comportamento que você deve contar.

Como tal, use:

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

em vez de; isso usará testes de igualdade ( ==) para determinar se faz responsereferência a uma sequência com o mesmo conteúdo (valor).

O mesmo se aplica a response == 'y' or 'yes'; use em response in ('y', 'yes')vez disso.