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()
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)
def test_diagonal() -> None:
diagonal_grid = ('2.............62....1....7...6..8...3...9...7...6..4...4'
'....8....52.............3')
sbrd = SB(diagonal_mode=True)
board = sbrd.grid_values(diagonal_grid)
result = sbrd.search(board)
assert result == solved_diag_sudoku
def test_reduce_puzzle_fail() -> None:
"""Given an empty board reduce will fail and return none"""
board_string = '.' * SB.num_boxes()
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.reduce_puzzle(board)
assert solution is None
def test_validate_fail() -> None:
"""Validation function correctly fails on incomplete board"""
board_string = ('4.....8.5.3..........7......2.....6.....8.4......1.......'
'6.3.7.5..2.....1.4......')
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.reduce_puzzle(board)
valid = sbrd.validate(solution)
assert valid is False
def test_diagonal() -> None:
"""Solve problem with diagonal constraint enabled"""
diagonal_grid = ('2.............62....1....7...6..8...3...9...7...6..4...4'
'....8....52.............3')
sbrd = SB(diagonal_mode=True)
board = sbrd.grid_values(diagonal_grid)
result = sbrd.search(board)
assert result == solved_diag_sudoku
def test_naked_multi_4() -> None:
quad_board = naked_multi_board_1.copy()
quad_board['A1'] = quad_board['A1'] + '2' # give it a 2 to clear
quad_board['A5'] = quad_board['A5'] + '2' # give it a 2 to clear
quad_board['F1'] = quad_board['F1'] + '1' # give it a 1 to clear
sbrd = SB()
board = sbrd.naked_multi(quad_board, 4)
assert board == naked_multi_board_1
def test_validate() -> None:
"""Validation function correctly validates complete board"""
board_string = ('..3.2.6..9..3.5..1..18.64....81.29..7.......8..67.82..'
'..26.95..8..2.3..9..5.1.3..')
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.reduce_puzzle(board)
valid = sbrd.validate(solution)
assert valid is True
def test_naked_multi_3() -> None:
"""Solve first naked twin board with naked triplets"""
triplet_board = naked_multi_board_1.copy()
triplet_board['A5'] = triplet_board['A5'] + '3' # give it a 3 to clear
triplet_board['I2'] = triplet_board['I2'] + '7' # give it a 7 to clear
triplet_board['G2'] = triplet_board['G2'] + '3' # give it a 3 to clear
sbrd = SB()
board = sbrd.naked_multi(triplet_board, 3)
assert board == naked_multi_board_1
def test_naked_multi_4() -> None:
"""Solve first naked twin board with naked quads"""
quad_board = naked_multi_board_1.copy()
# add some numbers to the board that the existing quads can clear
quad_board['A1'] = quad_board['A1'] + '2' # give it a 2 to clear
quad_board['A5'] = quad_board['A5'] + '2' # give it a 2 to clear
quad_board['F1'] = quad_board['F1'] + '1' # give it a 1 to clear
sbrd = SB()
board = sbrd.naked_multi(quad_board, 4)
assert board == naked_multi_board_1
def test_naked_multi_3() -> None:
"""Solve first naked twin board with naked triplets"""
triplet_board = naked_multi_board_1.copy()
# add some numbers to the board that the existing triples can clear
triplet_board['A5'] = triplet_board['A5'] + '3' # give it a 3 to clear
triplet_board['I2'] = triplet_board['I2'] + '7' # give it a 7 to clear
triplet_board['G2'] = triplet_board['G2'] + '3' # give it a 3 to clear
sbrd = SB()
board = sbrd.naked_multi(triplet_board, 3)
assert board == naked_multi_board_1
def test_search() -> None:
"""Ensure recursive search succeeds on supplied problem"""
board_string = ('4.....8.5.3..........7......2.....6.....8.4......1.......'
'6.3.7.5..2.....1.4......')
expected_string = ('4,1,7,3,6,9,8,2,5,6,3,2,1,5,8,9,4,7,9,5,8,7,2,4,3,1,6,'
'8,2,5,4,3,7,1,6,9,7,9,1,5,8,6,4,3,2,3,4,6,9,1,2,7,5,8,'
'2,8,9,6,4,3,5,7,1,5,7,3,2,9,1,6,8,4,1,6,4,8,7,5,2,9,3')
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.search(board)
SB.display(solution)
solution_string = SB.board_to_str(solution)
assert solution_string == expected_string
def test_reduce_puzzle() -> None:
"""Ensure board state after recursive reduce is correct"""
board_string = ('..3.2.6..9..3.5..1..18.64....81.29..7.......8..67.82..'
'..26.95..8..2.3..9..5.1.3..')
expected_string = ('4,8,3,9,2,1,6,5,7,9,6,7,3,4,5,8,2,1,2,5,1,8,7,6,4,9,3,'
'5,4,8,1,3,2,9,7,6,7,2,9,5,6,4,1,3,8,1,3,6,7,9,8,2,4,5,'
'3,7,2,6,8,9,5,1,4,8,1,4,2,5,3,7,6,9,6,9,5,4,1,7,3,8,2')
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.reduce_puzzle(board)
solution_string = sbrd.board_to_str(solution)
assert solution_string == expected_string
def test_only_choice() -> None:
"""Ensure board state after single only choice is correct"""
board_string = ('..3.2.6..9..3.5..1..18.64....81.29..7.......8..67.82..'
'..26.95..8..2.3..9..5.1.3..')
expected_string = ('45,8,3,9,2,1,6,5789,57,9,6,47,3,4,5,8,278,1,2,257,1,8,'
'7,6,4,23579,2357,345,345,8,1,3456,2,9,34567,34567,7,2,'
'9,5,34569,4,1,13456,8,1345,13459,6,7,3459,8,2,1345,'
'345,134,1347,2,6,8,9,5,1478,47,8,1467,47,2,5,3,17,6,9,'
'6,9,5,4,1,7,3,8,2')
board_dict = SB.grid_values(board_string)
sbrd = SB()
eliminated_board = sbrd.eliminate(board_dict)
only_choice_board = sbrd.only_choice(eliminated_board)
only_choice_string = SB.board_to_str(only_choice_board)
assert only_choice_string == expected_string
def test_eliminate() -> None:
"""Ensure board state after single elimination is correct"""
board_string = ('..3.2.6..9..3.5..1..18.64....81.29..7.......8..67.82..'
'..26.95..8..2.3..9..5.1.3..')
# eliminated board in comma separated format for easy comparison
expected_string = ('45,4578,3,49,2,147,6,5789,57,9,24678,47,3,47,5,78,278,'
'1,25,257,1,8,79,6,4,23579,2357,345,345,8,1,3456,2,9,'
'34567,34567,7,123459,49,459,34569,4,1,13456,8,1345,'
'13459,6,7,3459,8,2,1345,345,134,1347,2,6,478,9,5,1478,'
'47,8,1467,47,2,457,3,17,1467,9,46,4679,5,4,1,47,3,'
'24678,2467')
board = SB.grid_values(board_string)
eliminated_board = SB().eliminate(board)
eliminated_string = SB.board_to_str(eliminated_board)
assert eliminated_string == expected_string
def test_sorted_box_possibilities() -> None:
"""Ensure order of sorted box possibilities with given reduced board"""
board_string = ('4.....8.5.3..........7......2.....6.....8.4......1.......'
'6.3.7.5..2.....1.4......')
expected_sort = ['G2', 'H7', 'A4', 'A6', 'E4', 'F4', 'G1', 'G3', 'G5',
'H5', 'H8', 'I4', 'I5', 'A2', 'A5', 'A8', 'B8', 'C1',
'D1', 'D4', 'D6', 'E1', 'G7', 'H2', 'H9', 'I2', 'I6',
'A3', 'B1', 'B4', 'B5', 'B7', 'C2', 'C7', 'C8', 'D3',
'D5', 'D7', 'D9', 'E2', 'E3', 'E6', 'E8', 'E9', 'F1',
'F3', 'F6', 'F7', 'F8', 'F9', 'G9', 'I7', 'I8', 'I9',
'B6', 'B9', 'C3', 'C5', 'C6', 'C9', 'B3']
board = SB.grid_values(board_string)
sbrd = SB()
reduced = sbrd.reduce_puzzle(board)
SB.display(reduced)
sorted_boxes = SB.sorted_box_possibilities(reduced)
assert sorted_boxes == expected_sort
def test_reduce_puzzle_incomplete() -> None:
"""Ensure given a more complex puzzle reduce terminates incomplete"""
board_string = ('4.....8.5.3..........7......2.....6.....8.4......1.......'
'6.3.7.5..2.....1.4......')
expected_string = ('4,1679,12679,139,2369,269,8,1239,5,26789,3,1256789,'
'14589,24569,245689,12679,1249,124679,2689,15689,'
'125689,7,234569,245689,12369,12349,123469,3789,2,'
'15789,3459,34579,4579,13579,6,13789,3679,15679,15679,'
'359,8,25679,4,12359,12379,36789,4,56789,359,1,25679,'
'23579,23589,23789,289,89,289,6,459,3,1259,7,12489,5,'
'6789,3,2,479,1,69,489,4689,1,6789,4,589,579,5789,'
'23569,23589,23689')
board = SB.grid_values(board_string)
sbrd = SB()
solution = sbrd.reduce_puzzle(board)
solution_string = sbrd.board_to_str(solution)
assert solution_string == expected_string
def solve(board_string: str) -> BoardState:
"""Call search with diagonal_mode True in board.py"""
sbrd = SB(diagonal_mode=True)
board = sbrd.grid_values(board_string)
return sbrd.search(board)
def test_naked_multi_2() -> None:
"""Solve first naked twin problem with naked multi twins"""
sbrd = SB()
board = sbrd.naked_multi(naked_multi_board_1, 2)
assert board in naked_multi_solutions_1
def test_naked_twins_2() -> None:
sbrd = SB()
board = sbrd.naked_twins(before_naked_twins_2)
assert board in possible_solutions_2
def test_naked_twins_1() -> None:
"""Solve first naked twin problem"""
sbrd = SB()
board = sbrd.naked_twins(before_naked_twins_1)
assert board in possible_solutions_1
def test_naked_twins_2() -> None:
"""Solve second naked twin problem"""
sbrd = SB()
board = sbrd.naked_twins(before_naked_twins_2)
assert board in possible_solutions_2
def naked_twins(board_dict: BoardState) -> BoardState:
"""Call naked_twins in board.py"""
sbrd = SB()
return sbrd.naked_twins(board_dict)