Part 2

1 files changed, 13 insertions, 1 deletions

diff --git a/day5/test_init.py b/day5/test_init.py
index 5096a60..a4d7bff 100644
--- a/day5/test_init.py
+++ b/day5/test_init.py
@@ -7,7 +7,19 @@ def test_points_in_vector():
     ]
 
     assert points_in_vector(((9,7), (7,7))) == [
-        (7, 7), (8, 7), (9, 7)
+        (9, 7), (8, 7), (7, 7)
+    ]
+
+    assert points_in_vector(((1,1), (3,3)), False) == []
+
+    assert points_in_vector(((1,1), (3,3)), True) == [
+        (1, 1), (2, 2), (3, 3)
+    ]
+
+    assert points_in_vector(((9,7), (7,9)), False) == []
+
+    assert points_in_vector(((9,7), (7,9)), True) == [
+        (9, 7), (8, 8), (7, 9),
     ]