def test_matching_solutions(self):
row = brute_force.BruteForceRowSolver([2, 1], [0] * 5)
solution1 = [-1, 1, 1, -1, 1]
solution2 = [1, 1, -1, -1, 1]
expected_solution = [0, 1, 0, -1, 1]
solution = row._get_matching_solution(solution1, solution2)
self.assertEqual(expected_solution, solution)
def test_get_all_solutions_empty(self):
row = brute_force.BruteForceRowSolver([], [0] * 5)
expected_solutions = [
[-1, -1, -1, -1, -1],
]
solutions = row._get_all_solutions()
self.assertEqual(solutions, expected_solutions)
def test_get_all_solutions_three_clues(self):
row = brute_force.BruteForceRowSolver([2, 2, 3], [0] * 10)
solution = row._get_all_solutions()
expected_solution = [[1, 1, -1, 1, 1, -1, 1, 1, 1, -1],
[1, 1, -1, 1, 1, -1, -1, 1, 1, 1],
[1, 1, -1, -1, 1, 1, -1, 1, 1, 1],
[-1, 1, 1, -1, 1, 1, -1, 1, 1, 1]]
self.assertEqual(expected_solution, solution)
def test_row_brute_force_empty_values(self):
row = brute_force.BruteForceRowSolver([2, 1], [0] * 5)
changed = row.solve_brute_force()
self.assertTrue(changed)
self.assertEqual(row.values, [0, 1, 0, 0, 0])
def test_check_solutions(self):
row = brute_force.BruteForceRowSolver([2, 1], [1, 0, 0, 0, 0])
valid = row._check_solution([1, 1, 0, 1, 0])
self.assertTrue(valid)
valid = row._check_solution([0, 1, 1, 0, 1])
self.assertFalse(valid)
def test_number_of_solutions(self):
row = brute_force.BruteForceRowSolver([2, 1], [0] * 5)
no_solutions = row._get_number_of_solutions()
self.assertEqual(no_solutions, 3)
def test_number_of_solutions_large(self):
row = brute_force.BruteForceRowSolver([1, 1, 1], [0] * 10)
no_solutions = row._get_number_of_solutions()
self.assertEqual(no_solutions, 56)