def test_move_outcome_multiple_lines(self): """Test when lining up three jewels makes more line up, that we iterate until there are no more lines to remove. """ grid = [ ['green', 'yellow', 'purple', 'green', 'white', 'yellow', 'blue', 'yellow'], ['yellow', 'red', 'yellow', 'blue', 'orange', 'white', 'yellow', 'yellow'], ['blue', 'purple', 'orange', 'white', 'purple', 'blue', 'blue', 'red'], ['green', 'purple', 'red', 'yellow', 'green', 'white', 'yellow', 'yellow'], ['blue', 'green', 'red', 'white', 'purple', 'orange', 'white', 'yellow'], ['yellow', 'white', 'white', 'blue', 'orange', 'white', 'yellow', 'red'], ['purple', 'green', 'blue', 'red', 'yellow', 'blue', 'yellow', 'purple'], ['blue', 'green', 'yellow', 'orange', 'red', 'green', 'red', 'red']] final_grid = get_grid_after_move(grid, ((5, 4), (6, 4))) expected_final_grid = [ ['green', 'yellow', 'purple', 'green', 'white', None, None, None], ['yellow', 'red', 'yellow', 'blue', 'orange', None, 'blue', 'yellow'], ['blue', 'purple', 'orange', 'white', 'purple', None, 'yellow', 'yellow'], ['green', 'purple', 'red', 'yellow', 'green', None, 'blue', 'red'], ['blue', 'green', 'red', 'white', 'purple', 'white', 'orange', 'yellow'], ['yellow', 'white', 'white', 'blue', 'orange', 'blue', 'yellow', 'red'], ['purple', 'green', 'blue', 'red', 'yellow', 'blue', 'yellow', 'purple'], ['blue', 'green', 'yellow', 'orange', 'red', 'green', 'red', 'red']] self.assertEqual(final_grid, expected_final_grid)
def test_move_outcome(self): bitmap = autopy.bitmap.Bitmap.open("sample_images/board2.png") grid = get_current_grid(bitmap) final_grid = get_grid_after_move(grid, ((6, 2), (7, 2))) expected_final_grid = [ ['purple', 'green', 'green', 'yellow', 'white', 'blue', None, 'yellow'], ['white', 'blue', 'red', 'blue', 'purple', 'orange', None, 'white'], ['purple', 'purple', 'orange', 'yellow', 'white', 'blue', None, 'orange'], ['red', 'green', 'blue', 'purple', 'red', 'yellow', 'purple', 'blue'], ['blue', 'yellow', 'blue', 'yellow', 'orange', 'white', 'green', 'red'], ['orange', 'yellow', 'red', 'green', 'red', 'yellow', 'yellow', 'purple'], ['purple', 'green', 'white', 'blue', 'orange', 'orange', 'blue', 'purple'], ['blue', 'red', 'white', 'purple', 'yellow', 'orange', 'green', 'green']] self.assertEqual(expected_final_grid, final_grid)