# -*- coding: utf-8 -*-
from day20 import AssignmentOne


def test_rotation():
    items = [1, 2, -3, 3, -2, 0, 4]

    items = AssignmentOne.move(items, 1)
    assert items == [2, 1, -3, 3, -2, 0, 4]

    items = AssignmentOne.move(items, 2)
    assert items == [1, -3, 2, 3, -2, 0, 4]

    items = AssignmentOne.move(items, -3)
    assert items == [1, 2, 3, -2, -3, 0, 4]

    items = AssignmentOne.move(items, 3)
    assert items == [1, 2, -2, -3, 0, 3, 4]

    items = AssignmentOne.move(items, -2)
    assert items == [1, 2, -3, 0, 3, 4, -2]

    items = AssignmentOne.move(items, 0)
    assert items == [1, 2, -3, 0, 3, 4, -2]

    items = AssignmentOne.move(items, 4)
    assert items == [1, 2, -3, 4, 0, 3, -2]


def test_get_nth_number_after_0():
    items = [1, 2, -3, 4, 0, 3, -2]

    assert AssignmentOne.get_nth_number_after_0(items, 1000) == 4
    assert AssignmentOne.get_nth_number_after_0(items, 2000) == -3
    assert AssignmentOne.get_nth_number_after_0(items, 3000) == 2