mirror of
https://github.com/bertptrs/adventofcode.git
synced 2025-12-25 21:00:31 +01:00
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
import itertools
|
|
from typing import List
|
|
|
|
import pytest
|
|
|
|
from aoc2019.intcode import Computer
|
|
|
|
|
|
@pytest.mark.parametrize('program,expected', [
|
|
([1, 9, 10, 3, 2, 3, 11, 0, 99, 30, 40, 50], [3500, 9, 10, 70, 2, 3, 11, 0, 99, 30, 40, 50]),
|
|
([1, 0, 0, 0, 99], [2, 0, 0, 0, 99]),
|
|
([2, 3, 0, 3, 99], [2, 3, 0, 6, 99]),
|
|
([2, 4, 4, 5, 99, 0], [2, 4, 4, 5, 99, 9801]),
|
|
([1, 1, 1, 4, 99, 5, 6, 0, 99], [30, 1, 1, 4, 2, 5, 6, 0, 99]),
|
|
# This is technically part of day 5 but it fits the pattern
|
|
([1002, 4, 3, 4, 33], [1002, 4, 3, 4, 99])
|
|
])
|
|
def test_instructions_day2(program: List[int], expected: List[int]) -> None:
|
|
computer = Computer(program)
|
|
|
|
computer.run()
|
|
|
|
assert computer.program == expected
|
|
|
|
|
|
@pytest.mark.parametrize('number,program', itertools.product([7, 8, 9], [
|
|
[3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8],
|
|
[3, 3, 1108, -1, 8, 3, 4, 3, 99],
|
|
]))
|
|
def test_equality_opcode(program: List[int], number: int):
|
|
computer = Computer(program.copy())
|
|
computer.input.append(number)
|
|
|
|
computer.run()
|
|
|
|
assert computer.output.pop() == int(number == 8)
|
|
|
|
|
|
@pytest.mark.parametrize('number,program', itertools.product([7, 8, 9], [
|
|
[3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8],
|
|
[3, 3, 1107, -1, 8, 3, 4, 3, 99],
|
|
]))
|
|
def test_less_than_opcode(program: List[int], number: int):
|
|
computer = Computer(program.copy())
|
|
computer.input.append(number)
|
|
|
|
computer.run()
|
|
|
|
assert computer.output.pop() == int(number < 8)
|
|
|
|
|
|
@pytest.mark.parametrize('inputs,expected', [
|
|
(12, 1001),
|
|
(8, 1000),
|
|
(2, 999),
|
|
])
|
|
def test_day5_example(inputs: int, expected: int):
|
|
computer = Computer([3, 21, 1008, 21, 8, 20, 1005, 20, 22, 107, 8, 21, 20, 1006, 20, 31,
|
|
1106, 0, 36, 98, 0, 0, 1002, 21, 125, 20, 4, 20, 1105, 1, 46, 104,
|
|
999, 1105, 1, 46, 1101, 1000, 1, 20, 4, 20, 1105, 1, 46, 98, 99])
|
|
|
|
computer.input.append(inputs)
|
|
|
|
computer.run()
|
|
|
|
assert computer.output.pop() == expected
|
|
|
|
|
|
@pytest.mark.parametrize('program,output', [
|
|
([109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99],
|
|
[109, 1, 204, -1, 1001, 100, 1, 100, 1008, 100, 16, 101, 1006, 101, 0, 99]),
|
|
([1102, 34915192, 34915192, 7, 4, 7, 99, 0], [1219070632396864]),
|
|
([104, 1125899906842624, 99], [1125899906842624]),
|
|
])
|
|
def test_instructions_day9(program: List[int], output: List[int]) -> None:
|
|
computer = Computer(program)
|
|
computer.run()
|
|
|
|
result = list(computer.output)
|
|
|
|
assert result == output
|