test_aoc201712.py

"""Tests for AoC 12, 2017: Digital Plumber."""

# Standard library imports
import pathlib

# Third party imports
import aoc201712
import pytest

PUZZLE_DIR = pathlib.Path(__file__).parent


@pytest.fixture
def example1():
    puzzle_input = (PUZZLE_DIR / "example1.txt").read_text().strip()
    return aoc201712.parse_data(puzzle_input)


@pytest.fixture
def example2():
    puzzle_input = (PUZZLE_DIR / "example2.txt").read_text().strip()
    return aoc201712.parse_data(puzzle_input)


def test_parse_example1(example1):
    """Test that input is parsed properly."""
    assert example1 == {
        0: [2],
        1: [1],
        2: [0, 3, 4],
        3: [2, 4],
        4: [2, 3, 6],
        5: [6],
        6: [4, 5],
    }


def test_part1_example1(example1):
    """Test part 1 on example input."""
    assert aoc201712.part1(example1) == 6


def test_part2_example1(example1):
    """Test part 2 on example input."""
    assert aoc201712.part2(example1) == 2