def test_transform_index_pass(self):
size = 9
actions = [-1, size**size, None]
for action in actions:
with pytest.raises(AssertionError):
transform_index(action, THTransformation.diagonal_right_down,
size)
def test_transform_index_flipud(self):
size = 9
action = (0, 0)
expected_action = move_to_index((8, 0), size)
actual_action = transform_index(move_to_index(action, size),
THTransformation.flipud, size)
assert expected_action == actual_action
def test_transform_index_rotate_270(self):
size = 9
action = (0, 0)
expected_action = move_to_index((0, 8), size)
actual_action = transform_index(move_to_index(action, size),
THTransformation.rotate_270, size)
assert expected_action == actual_action
def test_transform_index_diagonal_right_down(self):
size = 9
action = (0, 0)
expected_action = move_to_index((0, 0), size)
actual_action = transform_index(move_to_index(action, size),
THTransformation.diagonal_right_down,
size)
assert expected_action == actual_action