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)
