Solve day 5

05/puzzle_test.py

@@ -0,0 +1,75 @@
+from puzzle import (RangeConversion, resolve_destination, resolve_source,
+                    solve_pt_1, solve_pt_2)
+
+mock_seeds = [79, 14, 55, 13]
+mock_destinations = [81, 14, 57, 13]
+mock_maps = [
+    [
+        RangeConversion(50, 98, 2),
+        RangeConversion(52, 50, 48),
+    ],
+    [
+        RangeConversion(0, 15, 37),
+        RangeConversion(37, 52, 2),
+        RangeConversion(39, 0, 15),
+    ],
+    [
+        RangeConversion(49, 53, 8),
+        RangeConversion(0, 11, 42),
+        RangeConversion(42, 0, 7),
+        RangeConversion(57, 7, 4),
+    ],
+    [
+        RangeConversion(88, 18, 7),
+        RangeConversion(18, 25, 70),
+    ],
+    [
+        RangeConversion(45, 77, 23),
+        RangeConversion(81, 45, 19),
+        RangeConversion(68, 64, 13),
+    ],
+    [
+        RangeConversion(0, 69, 1),
+        RangeConversion(1, 0, 69),
+    ],
+    [
+        RangeConversion(60, 56, 37),
+        RangeConversion(56, 93, 4),
+    ],
+]
+
+
+def test_range_convevrsion_resolve():
+    r = mock_maps[0][0]
+    assert r.resolve(97) is None
+    assert r.resolve(98) == 50
+    assert r.resolve(99) == 51
+    assert r.resolve(100) is None
+
+
+def test_range_conversion_reverse_resolve():
+    r = mock_maps[0][0]
+    assert r.reverse_resolve(49) is None
+    assert r.reverse_resolve(50) == 98
+    assert r.reverse_resolve(51) == 99
+    assert r.reverse_resolve(52) is None
+
+
+def test_resolve_destination():
+    assert [
+        resolve_destination(seed, mock_maps[0]) for seed in mock_seeds
+    ] == mock_destinations
+
+
+def test_resolve_source():
+    assert [
+        resolve_source(dest, mock_maps[0]) for dest in mock_destinations
+    ] == mock_seeds
+
+
+def test_solve_1():
+    assert solve_pt_1(mock_seeds, mock_maps) == 35
+
+
+def test_solve_2():
+    assert solve_pt_2(mock_seeds, mock_maps) == 46