test_task1.py

import pytest
from helpers.test_helpers import *


@pytest.mark.parametrize('player, expected', [
    (0, 1),
    (1, 0),
])
def test_get_opponent(player, expected):
    assert task1.get_opponent(player) == expected


@pytest.mark.parametrize('invalid_player', [
    -1,
    -2,
    2,
    3,
    1000,
    3.1415,
    9
])
def test_get_opponent_invalid_does_not_crash(invalid_player):
    task1.get_opponent(invalid_player)