test_task6.py

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

@pytest.mark.parametrize('size, expected', [
    (i, i) for i in range(100)
])

def test_board_size(size, expected):
    board = task6.make_boolean_square_board(size)
    assert len(board) == expected, f"Not all rows are of length {expected} in the board returned by make_boolean_square_board({size})."
    assert all(len(row) == expected for row in board), f"Not all columns are of length {expected} in the board returned by make_boolean_square_board({size})."
    assert all(cell == False for row in board for cell in row), f"Not all cells are set to False in the board returned by make_boolean_square_board({size})."