day 1 - part 2

tests/2024/day1/test_common.py

@@ -0,0 +1,15 @@
+from pathlib import Path
+
+from day1 import common
+
+TEST_FILE = Path(__file__).parent / "test-data"
+
+
+def test_get_lists():
+    assert common.get_lists(TEST_FILE)[0] == [3, 4, 2, 1, 3, 3]
+    assert common.get_lists(TEST_FILE)[1] == [4, 3, 5, 3, 9, 3]
+
+
+def test_get_sorted_lists():
+    assert common.get_ordered_lists(TEST_FILE)[0] == [1, 2, 3, 3, 3, 4]
+    assert common.get_ordered_lists(TEST_FILE)[1] == [3, 3, 3, 4, 5, 9]

tests/2024/day1/test_part1.py

@@ -2,13 +2,13 @@ from pathlib import Path
 
 from day1 import part1
 
-TEST_FILE = Path(__file__).parent / "test-data"
+
+def test_compute_distance():
+    list1 = [1, 2, 3, 3, 3, 4]
+    list2 = [3, 3, 3, 4, 5, 9]
+    assert part1.compute_distance(list1, list2) == 11
 
 
-def test_first():
-    assert part1.compute(TEST_FILE) == 11
-
-
-def test_get_lists():
-    assert part1.get_ordered_lists(TEST_FILE)[0] == [1, 2, 3, 3, 3, 4]
-    assert part1.get_ordered_lists(TEST_FILE)[1] == [3, 3, 3, 4, 5, 9]
+def test_main():
+    test_file = Path(__file__).parent / "test-data"
+    assert part1.main(test_file) == 11

tests/2024/day1/test_part2.py

@@ -0,0 +1,14 @@
+from pathlib import Path
+
+from day1 import part2
+
+
+def test_compute_similarity_score():
+    dict1 = {1: 1, 2: 1, 3: 3, 4: 1}
+    dict2 = {3: 3, 4: 1, 5: 1, 9: 1}
+    assert part2.compute_similarity_score(dict1, dict2) == 31
+
+
+def test_main():
+    test_file = Path(__file__).parent / "test-data"
+    assert part2.main(test_file) == 31