1 files changed, 22 insertions, 0 deletions

diff --git a/day3/test_init.py b/day3/test_init.py
new file mode 100644
index 0000000..ae4033e
--- /dev/null
+++ b/day3/test_init.py
@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+import pytest
+
+from day3 import AssignmentOne, AssignmentTwo, Assignment
+
+
+class TestAssignment:
+    data = [
+        ["987654321111111", 2, 98],
+        ["811111111111119", 2, 89],
+        ["234234234234278", 2, 78],
+        ["818181911112111", 2, 92],
+        ["987654321111111", 12, 987654321111],
+        ["811111111111119", 12, 811111111119],
+        ["234234234234278", 12, 434234234278],
+        ["818181911112111", 12, 888911112111],
+    ]
+
+    @pytest.mark.parametrize("battery,n,joltage", data)
+    def test_find_highest_joltage(self, battery: str, n: int, joltage: int):
+        banks = next(Assignment.parse_item(battery))
+        assert Assignment.find_highest_joltage(banks, n) == joltage