day 4 - part 2

tests/aoc_2024/day4/test_part2.py

@@ -0,0 +1,33 @@
+from pathlib import Path
+
+from aoc_2024.day4 import part2
+
+import numpy as np
+
+
+def test_find_all():
+    data_str = """MMMSXXMASM
+MSAMXMSMSA
+AMXSXMAAMM
+MSAMASMSMX
+XMASAMXAMM
+XXAMMXXAMA
+SMSMSASXSS
+SAXAMASAAA
+MAMMMXMMMM
+MXMXAXMASX"""
+    data = np.full((10, 10), "")
+    i, j = 0, 0
+    for character in data_str:
+        if character != "\n":
+            data[i, j] = character
+            j += 1
+        else:
+            i += 1
+            j = 0
+    assert part2.find_x_mas(data) == 9
+
+
+def test_main():
+    data_file = Path(__file__).parent / "test-data"
+    assert part2.main(data_file) == 9