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
28
29
30
31
32
33
34
35
|
# -*- 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
|
