Solve day 3

2023-12-09 19:33:23 -08:00
parent 5a938d3f44
commit 637b097245
6 changed files with 374 additions and 0 deletions
--- a/03/schematic_test.py
+++ b/03/schematic_test.py
@@ -0,0 +1,87 @@
+from schematic import Schematic, SchematicNumber, SchematicSymbol
+
+mock_input = """
+467..114..
+...*......
+..35..633.
+......#...
+617*......
+.....+.58.
+..592.....
+......755.
+...$.*....
+.664.598..
+""".strip()
+
+
+def test_schematic_number_extend_digit():
+    sn = SchematicNumber("", 1, 2)
+    assert sn.number == ""
+    sn = sn.extend_digit("3")
+    assert sn.number == "3"
+    sn = sn.extend_digit("5")
+    assert sn.number == "35"
+    sn = sn.extend_digit("1")
+    assert sn.number == "351"
+
+
+def test_parse_schematic():
+    assert Schematic.parse(mock_input) == Schematic(
+        [
+            SchematicNumber("467", 0, 0),
+            SchematicNumber("114", 0, 5),
+            SchematicNumber("35", 2, 2),
+            SchematicNumber("633", 2, 6),
+            SchematicNumber("617", 4, 0),
+            SchematicNumber("58", 5, 7),
+            SchematicNumber("592", 6, 2),
+            SchematicNumber("755", 7, 6),
+            SchematicNumber("664", 9, 1),
+            SchematicNumber("598", 9, 5),
+        ],
+        [
+            SchematicSymbol("*", 1, 3),
+            SchematicSymbol("#", 3, 6),
+            SchematicSymbol("*", 4, 3),
+            SchematicSymbol("+", 5, 5),
+            SchematicSymbol("$", 8, 3),
+            SchematicSymbol("*", 8, 5),
+        ],
+    )
+
+
+def test_schematic_part_numbers():
+    assert Schematic.parse(mock_input).part_numbers() == [
+        (
+            SchematicNumber("467", 0, 0),
+            SchematicSymbol("*", 1, 3),
+        ),
+        (
+            SchematicNumber("35", 2, 2),
+            SchematicSymbol("*", 1, 3),
+        ),
+        (
+            SchematicNumber("633", 2, 6),
+            SchematicSymbol("#", 3, 6),
+        ),
+        (
+            SchematicNumber("617", 4, 0),
+            SchematicSymbol("*", 4, 3),
+        ),
+        (
+            SchematicNumber("592", 6, 2),
+            SchematicSymbol("+", 5, 5),
+        ),
+        (
+            SchematicNumber("755", 7, 6),
+            SchematicSymbol("*", 8, 5),
+        ),
+        (
+            SchematicNumber("664", 9, 1),
+            SchematicSymbol("$", 8, 3),
+        ),
+        (
+            SchematicNumber("598", 9, 5),
+            SchematicSymbol("*", 8, 5),
+        ),
+    ]