Added day 5 wipmain

1 files changed, 23 insertions, 0 deletions

diff --git a/day5/test_init.py b/day5/test_init.py
new file mode 100644
index 0000000..018387b
--- /dev/null
+++ b/day5/test_init.py
@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+from unittest import result
+
+import pytest
+
+from day5 import AssignmentTwo
+
+
+class TestAssignmentTwo:
+    data = [
+        [
+            [range(3, 6), range(10, 15), range(16, 21), range(12, 19)],
+            [range(3, 6), range(10, 21)],
+        ],
+        [[range(2, 8), range(4, 6)], [range(2, 8)]],
+        [[range(4, 6), range(2, 8)], [range(2, 8)]],
+        [[range(1, 3), range(3, 6)], [range(1, 6)]],
+        [[range(3, 6), range(1, 3)], [range(1, 6)]],
+    ]
+
+    @pytest.mark.parametrize("ranges,result", data)
+    def test_merge_ranges(self, ranges: list[range], result: list[range]):
+        assert AssignmentTwo.merge_ranges(ranges) == result