1 files changed, 27 insertions, 0 deletions

diff --git a/day20/test_init.py b/day20/test_init.py
new file mode 100644
index 0000000..b3ccc44
--- /dev/null
+++ b/day20/test_init.py
@@ -0,0 +1,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]