pigeon-magnet-solver/tests/test_board.py

from pigeon_magnet_solver.board import Board


def test_default():
    board = Board.default_puzzle

    assert board.height == 4, "Wrong height"
    assert board.width == 3, "Wrong width"

    assert board.columns == ("001x", "1101", "110x"), "Wrong columns"
    assert board.binary_repr == (4, 11, 3), "Wrong binary repr"
    assert board.solution_class == (1, 3, 2), "Wrong solution class"

    assert (board2 := board.click(4)) is not None
    assert board2.binary_repr == (4, 14, 3)

    assert (board2 := board.click(5)) is not None
    assert board2.binary_repr == (4, 11, 6)

    for idx in (3, 7):
        assert board.click(idx) is None