test_aoc201704.py

"""Tests for AoC 4, 2017: High-Entropy Passphrases."""

# Standard library imports
import pathlib

# Third party imports
import aoc201704
import pytest

PUZZLE_DIR = pathlib.Path(__file__).parent


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


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


def test_parse_example1(example1):
    """Test that input is parsed properly."""
    assert example1 == [
        ["aa", "bb", "cc", "dd", "ee"],
        ["aa", "bb", "cc", "dd", "aa"],
        ["aa", "bb", "cc", "dd", "aaa"],
    ]


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


def test_part2_example2(example2):
    """Test part 2 on example input."""
    assert aoc201704.part2(example2) == 3