def test_2x2_up():
expected_output = [
[4, 2],
[8, 0]
]
_input = [
[2, 0],
[4, 8]
]
output = transform(_input, UP)
assert output == expected_output
def test_2x2_down_with_a_resulting_zero_row():
expected_output = [
[8, 0],
[0, 4]
]
_input = [
[0, 8],
[4, 0]
]
output = transform(_input, DOWN)
assert output == expected_output
def test_2x2_left():
expected_output = [
[8, 2],
[4, 0]
]
_input = [
[2, 8],
[0, 4]
]
output = transform(_input, LEFT)
assert output == expected_output
def test_2x2_right():
expected_output = [
[8, 2],
[4, 0]
]
_input = [
[8, 2],
[4, 0]
]
output = transform(_input, RIGHT)
assert output == expected_output
def test_2x2_down():
expected_output = [
[8, 0],
[2, 4]
]
_input = [
[2, 8],
[4, 0]
]
output = transform(_input, DOWN)
assert output == expected_output