def test_diagonal_units() -> None:
"""Ensure diagonal units are the correct ones"""
units = [['A1', 'B2', 'C3', 'D4', 'E5', 'F6', 'G7', 'H8', 'I9'],
['I1', 'H2', 'G3', 'F4', 'E5', 'D6', 'C7', 'B8', 'A9']]
# pylint: disable=protected-access
diagonal_units = SB._diagonal_units()
assert diagonal_units == units
def test_diagonal_board() -> None:
"""Ensure diagonal board mode boolean constructor works"""
sbrd = SB()
sbrd_diagonal = SB(diagonal_mode=True)
assert sbrd.all_units() != sbrd_diagonal.all_units()
# pylint: disable=protected-access
assert sbrd.all_units() + SB._diagonal_units() == sbrd_diagonal.all_units()
