import itertools
import os

from .comp import OpcodeComputer


inputfile = os.path.join(os.path.dirname(__file__), "input/day09_input")


def test_day9a():
    comp = OpcodeComputer(inputfile)
    res = list(comp.process_op([1]))
    assert res[0] == 3638931938


def test_day9b():
    comp = OpcodeComputer(inputfile)
    res = list(comp.process_op([2]))
    assert res[0] == 86025