Use pytest.mark.parametrize instead of test generator