advent_of_code_2020/day4/test_day4.py

66 lines
1.9 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 example1():
puzzle_input = (PUZZLE_DIR / "example1").read_text().strip()
return aoc.parse(puzzle_input)
@pytest.fixture
def example2():
puzzle_input = (PUZZLE_DIR / "example2").read_text().strip()
return aoc.parse(puzzle_input)
# @pytest.mark.skip(reason="Not implemented")
def test_parse_example1(example1):
"""Test that input is parsed properly"""
assert example1 == [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_example1(example1):
"""Test part 1 on example input"""
assert aoc.part1(example1) == 2
# @pytest.mark.skip(reason="Not implemented")
def test_part2_example2(example1):
"""Test part 2 on example input"""
assert aoc.part2(example1) == 2