advent_of_code_2020/day6/test_day6.py

56 lines
1.4 KiB
Python
Raw Normal View History

2022-09-14 00:44:46 +02:00
#! /usr/bin/env python3
import pathlib
import pytest
import day6 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 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 == [
[
['a', 'b', 'c']
],
[
['a'],
['b'],
['c']
],
[
['a', 'b'],
['a', 'c']
],
[
['a'],
['a'],
['a'],
['a']
],
[
['b']
]
]
# @pytest.mark.skip(reason="Not implemented")
def test_part1_example1(example1):
"""Test part 1 on example input"""
assert aoc.part1(example1) == 11
# @pytest.mark.skip(reason="Not implemented")
def test_part2_example2(example1):
"""Test part 2 on example input"""
assert aoc.part2(example1) == 6