import unittest
import os
from .comp import OpcodeComputer


class Day5(unittest.TestCase):
    inputfile = os.path.join(os.path.dirname(__file__), "input/day5_input")

    def test_day5a(self):
        sut = OpcodeComputer(self.inputfile)

        outp = list(sut.process_op([1]))
        self.assertEqual(int(outp[-1]), 8332629)

    def test_day5b(self):
        sut = OpcodeComputer(self.inputfile)

        outp = list(sut.process_op([5]))
        self.assertEqual(int(outp[0]), 8805067)