Day 3

3 files changed, 65 insertions, 0 deletions

diff --git a/day3/__init__.py b/day3/__init__.py
new file mode 100644
index 0000000..9025f4a
--- /dev/null
+++ b/day3/__init__.py
@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+from abc import ABC
+from typing import Iterator
+
+from aoc import BaseAssignment
+
+
+class Assignment(BaseAssignment, ABC):
+    @classmethod
+    def parse_item(cls, item: str) -> Iterator[list[int]]:
+        yield [int(i) for i in item]
+
+    @classmethod
+    def find_highest_joltage(cls, item: list[int], n: int) -> int:
+        values = list()
+        item_length = len(item)
+
+        last_index = -1
+        for end_index in range(item_length - n + 1, item_length + 1):
+            start_index = last_index + 1
+            value = max(item[start_index:end_index])
+            last_index = item.index(value, start_index, end_index)
+            values.append(str(value))
+
+        return int("".join(values))
+
+
+class AssignmentOne(Assignment):
+    example_result = 357
+
+    def run(self, input: Iterator[list[int]]) -> int:
+        return sum([self.find_highest_joltage(item, 2) for item in input])
+
+
+class AssignmentTwo(Assignment):
+    example_result = 3121910778619
+
+    def run(self, input: Iterator[list[int]]) -> int:
+        return sum([self.find_highest_joltage(item, 12) for item in input])
diff --git a/day3/example.txt b/day3/example.txt
new file mode 100644
index 0000000..7255fca
--- /dev/null
+++ b/day3/example.txt
@@ -0,0 +1,4 @@
+987654321111111
+811111111111119
+234234234234278
+818181911112111
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