test_task2.py

import pytest
from tests.helpers.test_helpers import *
from src import task2

@pytest.mark.parametrize('coord, board, expected', [
    ((0, 0), board3, True),
    ((0, 2), board3, True),
    ((2, 0), board3, True),
    ((1, 1), board3, True),
    ((0, 3), board3, False),
    ((3, 0), board3, False),
    ((-1, 2), board3, False),
    ((2, -1), board3, False),
    ((-1, -1), board3, False),
    ((0, 0), board10, True),
    ((0, 9), board10, True),
    ((9, 0), board10, True),
    ((5, 5), board10, True),
    ((10, 0), board10, False),
    ((0, 10), board10, False),
    ((-2, 9), board10, False),
    ((9, -2), board10, False),
    ((-2, -2), board10, False),
    ((0, 0), board0, False),
    ((0, 1), board0, False),
    ((1, 0), board0, False),
    ((0, -1), board0, False),
    ((-1, 0), board0, False),
])

def test_check_valid_coord(coord, board, expected):
    result = task2.check_valid_coord(coord, board)
    assert result == expected, f"go_to_next_round(check_valid_coord({coord}, boardX) with boardX of length {len(board)} returned {result}, but the expected value is {expected}."