test_task12.py

import pytest
from helpers.test_helpers import *

@pytest.mark.parametrize('player, current_round, expected_round', [
    (0, 1, 1),
    (1, 1, 2),
    (0, 2, 2),
    (1, 2, 3),
    (0, 5, 5),
    (1, 5, 6),
])
def test_go_to_next_round(player, current_round, expected_round):
    assert task12.go_to_next_round(player, current_round) == expected_round