1
0
Fork 0
advent-of-code/2020/days/04/test_day4.py

73 lines
1.8 KiB
Python

#! /usr/bin/env python3
import pathlib
import pytest
import day4 as aoc
from day4 import Passport
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.txt").read_text().strip()
return aoc.parse(puzzle_input)
def test_parse(example):
"""Test that input is parsed properly"""
assert example == [
Passport(
birthYear=1937,
issueYear=2017,
expirationYear=2020,
height="183cm",
hairColor="#fffffd",
eyeColor="gry",
passportId="860033327",
countryId="147",
),
Passport(
birthYear=1929,
issueYear=2013,
expirationYear=2023,
height=None,
hairColor="#cfa07d",
eyeColor="amb",
passportId="028048884",
countryId="350",
),
Passport(
birthYear=1931,
issueYear=2013,
expirationYear=2024,
height="179cm",
hairColor="#ae17e1",
eyeColor="brn",
passportId="760753108",
countryId=None,
),
Passport(
birthYear=None,
issueYear=2011,
expirationYear=2025,
height="59in",
hairColor="#cfa07d",
eyeColor="brn",
passportId="166559648",
countryId=None,
),
]
# @pytest.mark.skip(reason="Not implemented")
def test_part1(example):
"""Test part 1 on example input"""
assert aoc.part1(example) == 2
# @pytest.mark.skip(reason="Not implemented")
def test_part2(example):
"""Test part 2 on example input"""
assert aoc.part2(example) == 2