33 lines
595 B
Python
33 lines
595 B
Python
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
|