33 lines
886 B
Python
33 lines
886 B
Python
#! /usr/bin/env python3
|
|
|
|
import pathlib
|
|
import pytest
|
|
import day2 as aoc
|
|
|
|
PUZZLE_DIR = pathlib.Path(__file__).parent
|
|
|
|
|
|
# these test fixtures setup the test, mainly by reading the filename into a string in this simple case.
|
|
@pytest.fixture
|
|
def example():
|
|
puzzle_input = (PUZZLE_DIR / "example").read_text().strip()
|
|
return aoc.parse(puzzle_input)
|
|
|
|
|
|
def test_parse(example):
|
|
"""Test that input is parsed properly"""
|
|
assert example == [
|
|
aoc.PasswordSpec(first=1, second=3, letter="a", password="abcde"),
|
|
aoc.PasswordSpec(first=1, second=3, letter="b", password="cdefg"),
|
|
aoc.PasswordSpec(first=2, second=9, letter="c", password="ccccccccc"),
|
|
]
|
|
|
|
|
|
def test_part1(example):
|
|
"""Test part 1 on example input"""
|
|
assert aoc.part1(example) == 2
|
|
|
|
|
|
def test_part2(example):
|
|
"""Test part 2 on example input"""
|
|
assert aoc.part2(example) == 1
|