1 files changed, 41 insertions, 0 deletions

diff --git a/day2/test_init.py b/day2/test_init.py
new file mode 100644
index 0000000..dc1ab40
--- /dev/null
+++ b/day2/test_init.py
@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+import pytest
+
+from day2 import AssignmentOne, AssignmentTwo
+
+
+class TestAssignmentOne:
+    data = [
+        (range(11, 23), [11, 22]),
+        (range(95, 116), [99]),
+        (range(998, 1013), [1010]),
+        (range(1188511880, 1188511891), [1188511885]),
+        (range(222220, 222225), [222222]),
+        (range(1698522, 1698529), []),
+        (range(446443, 446450), [446446]),
+        (range(38593856, 38593863), [38593859]),
+    ]
+
+    @pytest.mark.parametrize("r,invalid", data)
+    def test_find_invalid_ids(self, r: range, invalid: list[str]):
+        assert list(AssignmentOne.find_invalid_ids(r)) == invalid
+
+
+class TestAssignmentTwo:
+    data = [
+        (range(11, 23), [11, 22]),
+        (range(95, 116), [99, 111]),
+        (range(998, 1013), [999, 1010]),
+        (range(1188511880, 1188511891), [1188511885]),
+        (range(222220, 222225), [222222]),
+        (range(1698522, 1698529), []),
+        (range(446443, 446450), [446446]),
+        (range(38593856, 38593863), [38593859]),
+        (range(565653, 565660), [565656]),
+        (range(824824821, 824824828), [824824824]),
+        (range(2121212118, 2121212125), [2121212121]),
+    ]
+
+    @pytest.mark.parametrize("r,invalid", data)
+    def test_find_invalid_ids(self, r: range, invalid: list[str]):
+        assert list(AssignmentTwo.find_invalid_ids(r)) == invalid