Source code for day16.tests.test_day16

"""Test day16 main functions."""
from typing import TYPE_CHECKING

from day16.day16 import INPUT_SMALL, part1, part2
from day16.lib.parsers import get_input

if TYPE_CHECKING:
    from day16.lib.world import World



[docs]
def test_part1() -> None:
    """Test part1."""
    world: World = get_input(INPUT_SMALL)
    assert part1(world) == 46




[docs]
def test_part2() -> None:
    """Test part2."""
    world: World = get_input(INPUT_SMALL)
    assert part2(world) == 51