diff --git a/game2048/grid_2048.py b/game2048/grid_2048.py index 9cb660b1e28b04ea6d1aa08eb0e9943b5dc98992..c6694f046173c0f5731c11efdcfddda5eac504b2 100644 --- a/game2048/grid_2048.py +++ b/game2048/grid_2048.py @@ -115,3 +115,6 @@ def move_row_left(row): row,n = del_zeros(row) new_row = move_left_bis(row) return new_row + [0]*(n - len(new_row)) + +def move_row_right(row): + return move_row_left(row[-1:])[-1:] diff --git a/game2048/test_grid_2048.py b/game2048/test_grid_2048.py index bb35758be7d31d34a6afc9ae5ed239349a7518bb..82a2333bd405fca0e671073760edc8f3362d34e1 100644 --- a/game2048/test_grid_2048.py +++ b/game2048/test_grid_2048.py @@ -72,3 +72,14 @@ def test_move_row_left(): assert move_row_left([2, 4, 2, 2]) == [2, 4, 4, 0] assert move_row_left([2, 4, 4, 0]) == [2, 8, 0, 0] assert move_row_left([4, 8, 16, 32]) == [4, 8, 16, 32] + +def test_move_row_right(): + assert move_row_right([2, 0, 0, 0]) == [0, 0, 0, 2] + assert move_row_right([0, 2, 0, 4]) == [0, 0, 2, 4] + assert move_row_right([2, 2, 0, 4]) == [0, 0, 4, 4] + assert move_row_right([2, 2, 2, 2]) == [0, 0, 4, 4] + assert move_row_right([4, 2, 0, 2]) == [0, 0, 4, 4] + assert move_row_right([2, 0, 0, 2]) == [0, 0, 0, 4] + assert move_row_right[2, 4, 2, 2]) == [0, 2, 4, 4] + assert move_row_right([2, 4, 4, 0]) == [0, 0, 2, 8] + assert move_row_right([4, 8, 16, 32]) == [4, 8, 16, 32]