Pytest, cycling through responses

test_cycle_through_responses.py

import sys
from itertools import cycle
from unittest import mock


def test_cycle_through_responses(monkeypatch):
    return_data = ["h", "hello"]
    test_mock = mock.Mock(side_effect=cycle(return_data))

    m = sys.modules[__name__]
    monkeypatch.setattr(m, "myfunc", test_mock)

    assert myfunc() == "h"
    assert myfunc() == "hello"
    assert myfunc() == "h"
    assert myfunc() == "hello"


def myfunc():
    return