import unittest
import os
from . import comp


class Day5(unittest.TestCase):
    def get_code(self):
        inputfile = os.path.join(os.path.dirname(__file__), "input/day5_input")
        with open(inputfile) as fp:
            return [int(k) for k in fp.readline().split(',')]

    def test_day5a(self):
        output_val = []

        sut = comp.OpcodeComputer()
        comp.input = lambda: 1
        comp.print = lambda s: output_val.append(s)

        sut.process_op(self.get_code())
        self.assertEqual(int(output_val[-1]), 8332629)

    def test_day5b(self):
        output_val = []

        sut = comp.OpcodeComputer()
        comp.input = lambda: 5
        comp.print = lambda s: output_val.append(s)

        sut.process_op(self.get_code())
        self.assertEqual(int(output_val[-1]), 8805067)