Python giả nhiều giá trị trả về

Tôi đang sử dụng pythons mock.patch và muốn thay đổi giá trị trả lại cho mỗi cuộc gọi. Đây là lời cảnh báo: chức năng được vá không có đầu vào, vì vậy tôi không thể thay đổi giá trị trả về dựa trên đầu vào.

Đây là mã của tôi để tham khảo.

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

Mã kiểm tra của tôi:

@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.promptchỉ là một phiên bản "đầu vào" độc lập trên nền tảng (python 2 và 3). Vì vậy, cuối cùng tôi đang cố gắng để chế nhạo đầu vào của người dùng. Tôi đã thử sử dụng một danh sách cho giá trị trả về, nhưng nó không hoạt động.

Bạn có thể thấy rằng nếu giá trị trả về là một cái gì đó không hợp lệ, tôi sẽ chỉ nhận được một vòng lặp vô hạn ở đây. Vì vậy, tôi cần một cách để cuối cùng thay đổi giá trị trả về, để thử nghiệm của tôi thực sự kết thúc.

(một cách khác có thể để trả lời câu hỏi này có thể là giải thích cách tôi có thể bắt chước đầu vào của người dùng trong bài kiểm tra đơn vị)

Không phải là một bản sao của câu hỏi này chủ yếu vì tôi không có khả năng thay đổi đầu vào.

Một trong những ý kiến ​​của Câu trả lời cho câu hỏi này là cùng một dòng, nhưng không có câu trả lời / nhận xét nào được cung cấp.

Bạn có thể gán một lần lặp cho side_effectvà giả sẽ trả về giá trị tiếp theo trong chuỗi mỗi lần nó được gọi:

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

Trích dẫn Mock()tài liệu :

Nếu side_effect là một iterable thì mỗi lệnh gọi tới mock sẽ trả về giá trị tiếp theo từ iterable.

Như một bên, thử nghiệm response is not 'y' or 'n' or 'yes' or 'no'sẽ không hoạt động; bạn đang hỏi xem biểu thức (response is not 'y')là đúng hay 'y'đúng (luôn luôn như vậy, một chuỗi không trống luôn luôn đúng), v.v ... Các biểu thức khác nhau ở hai bên của ortoán tử là độc lập . Xem Làm cách nào để kiểm tra một biến đối với nhiều giá trị?

Bạn cũng không nên sử dụng isđể kiểm tra đối với một chuỗi. Trình thông dịch CPython có thể sử dụng lại các đối tượng chuỗi trong một số trường hợp nhất định , nhưng đây không phải là hành vi bạn nên dựa vào.

Như vậy, sử dụng:

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

thay thế; điều này sẽ sử dụng các phép kiểm bằng ( ==) để xác định xem responsetham chiếu một chuỗi có cùng nội dung (giá trị) hay không.

Áp dụng tương tự cho response == 'y' or 'yes'; sử dụng response in ('y', 'yes')thay thế.