Day 9

1 files changed, 25 insertions, 0 deletions

diff --git a/day9/test_init.py b/day9/test_init.py
index eb889c7..431dbbb 100644
--- a/day9/test_init.py
+++ b/day9/test_init.py
@@ -75,6 +75,31 @@ class TestAssignment:
             assignment.visualize(range(1), range(-5, 1), unique_positions) == expected
         )
 
+    def test_diagonal(self):
+        expected = "\n".join(
+            [
+                "..#",
+                ".#.",
+                "s..",
+            ]
+        )
+
+        input = [
+            "R 1",
+            "U 1",
+            "R 1",
+            "U 1",
+            "R 1",
+            "U 1",
+            "R 1",
+            "U 1",
+        ]
+
+        assignment = day9.AssignmentOne(path=os.path.dirname(day9.__file__))
+        unique_positions = assignment.unique_tail_positions(input=input, length=2)
+
+        assert assignment.visualize(range(3), range(3), unique_positions) == expected
+
 
 class TestAssignmentOne:
     def test_output_visualization(self):