summaryrefslogtreecommitdiffstats
path: root/day20/test_init.py
blob: b3ccc4471886073a7bf94c7fb5277aceb6a9a163 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# -*- 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]