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")