pigeon-magnet-solver/tests/test_helpers.py

18 lines
722 B
Python

from pigeon_magnet_solver.helpers import (_fix_exes, _rotate_r, _valid_rotate,
rotate_r)
def test_rotate_r():
assert _rotate_r(list("1x01")) == list("11x0")
assert _rotate_r(list("11x0")) == list("011x")
assert _rotate_r(list("011x")) == list("x011")
assert _rotate_r(list("x011")) == list("1x01")
assert _valid_rotate(list("1x01"), list("11x0")) is False
assert _valid_rotate(list("1x01"), list("011x")) is False
assert _valid_rotate(list("1x01"), list("x011")) is True
assert _valid_rotate(list("1x01"), list("1x01")) is True
assert _fix_exes(list("1x01"), list("x011")) == list("0x11")
assert rotate_r(list("1x01")) == list("0x11")