def test_find_color(self):
board = GameBoard('tests/test_puzzle.txt')
board.read_board()
rect = Rectangle(board, Point(0, 0), Point(0, 0), 3)
rect.draw_rectangle()
rect = Rectangle(board, Point(0, 0), Point(2, 2), 4)
rect.find_color()
self.assertEqual(rect.color, 3)
def update_solution(self, first_point: Point, second_point: Point):
self.board.solution = deepcopy(self.solution_backup)
for i in range(self.board.rows):
for j in range(self.board.cols):
self.squares[i][j].color = self.squares_backup[i][j]
rect = Rectangle(self.board, first_point, second_point, 0)
rect.find_color()
color = (250, 190, 250) if rect.color == -1 \
else COLORS[rect.color % (len(COLORS) - 1)][0:3]
rect.draw_rectangle()
for x in range(rect.top_left.x, rect.bottom_right.x + 1):
for y in range(rect.top_left.y, rect.bottom_right.y + 1):
self.squares[self.board.cols - y - 1][x].color = color
