def test_peers() -> None: """Peers for non diagonal board for 'A1'""" peer = 'A1' a1_peers = ['A2', 'A3', 'A4', 'A5', 'A6', 'A7', 'A8', 'A9', 'B1', 'B2', 'B3', 'C1', 'C2', 'C3', 'D1', 'E1', 'F1', 'G1', 'H1', 'I1'] board = SB() assert board.peers(peer) == set(a1_peers)
def test_peers_diagonal() -> None: """Peers for diagonal board for 'A1'""" peer = 'A1' a1_peers = ['A2', 'A3', 'A4', 'A5', 'A6', 'A7', 'A8', 'A9', 'B1', 'B2', 'B3', 'C1', 'C2', 'C3', 'D1', 'E1', 'F1', 'G1', 'H1', 'I1', 'I9', 'H8', 'D4', 'E5', 'F6', 'G7'] board = SB(diagonal_mode=True) assert board.peers(peer) == set(a1_peers)