bert/adventofcode
1
0
Fork 0
mirror of https://github.com/bertptrs/adventofcode.git synced 2025-12-25 21:00:31 +01:00
Code Issues Projects Releases Wiki Activity

Implement 2019 day 12

Browse Source
This commit is contained in:
Bert Peters
2021-01-25 20:55:39 +01:00
parent 2c767d5996
commit f94661d7a8
2 changed files with 94 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
2019/tests/test_day12.py Normal file
View File

@@ -0,0 +1,25 @@
import pytest
import numpy # type: ignore
from aoc2019.day12 import simulate_moons, find_repetition
SAMPLES = [
numpy.array([[-1, 0, 2], [2, -10, -7], [4, -8, 8], [3, 5, -1]]),
numpy.array([[-8, -10, 0], [5, 5, 10], [2, -7, 3], [9, -8, -3]]),
]
@pytest.mark.parametrize('iterations,result,moons', [
(10, 179, numpy.copy(SAMPLES[0])),
(100, 1940, numpy.copy(SAMPLES[1])),
])
def test_kinetic_energy(moons: numpy.array, iterations: int, result: int) -> None:
assert simulate_moons(moons, iterations) == result
@pytest.mark.parametrize('outcome,moons', [
(2772, numpy.copy(SAMPLES[0])),
(4686774924, numpy.copy(SAMPLES[1])),
])
def test_repetition(moons: numpy.array, outcome: int) -> None:
assert find_repetition(moons) == outcome