1 files changed, 37 insertions, 0 deletions

diff --git a/day15/test_init.py b/day15/test_init.py
new file mode 100644
index 0000000..e5f10b0
--- /dev/null
+++ b/day15/test_init.py
@@ -0,0 +1,37 @@
+from day15 import AssignmentTwo
+
+
+def test_icrease_map():
+    assert AssignmentTwo.increase_map([[8]]) == [
+        [8, 9, 1, 2, 3],
+        [9, 1, 2, 3, 4],
+        [1, 2, 3, 4, 5],
+        [2, 3, 4, 5, 6],
+        [3, 4, 5, 6, 7],
+    ]
+
+    assert AssignmentTwo.increase_map([
+        [0],
+        [1]
+    ]) == [
+        [0, 1, 2, 3, 4],
+        [1, 2, 3, 4, 5],
+        [1, 2, 3, 4, 5],
+        [2, 3, 4, 5, 6],
+        [2, 3, 4, 5, 6],
+        [3, 4, 5, 6, 7],
+        [3, 4, 5, 6, 7],
+        [4, 5, 6, 7, 8],
+        [4, 5, 6, 7, 8],
+        [5, 6, 7, 8, 9]
+    ]
+
+    assert AssignmentTwo.increase_map([
+        [0, 1],
+    ]) == [
+        [0, 1, 1, 2, 2, 3, 3, 4, 4, 5],
+        [1, 2, 2, 3, 3, 4, 4, 5, 5, 6],
+        [2, 3, 3, 4, 4, 5, 5, 6, 6, 7],
+        [3, 4, 4, 5, 5, 6, 6, 7, 7, 8],
+        [4, 5, 5, 6, 6, 7, 7, 8, 8, 9]
+    ]