def test_addition_opcode(self):
prog = IntcodeComputer()
sample_op = [1, 0, 0, 0, 99]
expectation = [2, 0, 0, 0, 99]
result = prog.compute_sequence(sample_op)
self.assertEqual(result, expectation)
def test_another_multiplication_opcode(self):
prog = IntcodeComputer()
sample_op = [2, 4, 4, 5, 99, 0]
expectation = [2, 4, 4, 5, 99, 9801]
result = prog.compute_sequence(sample_op)
self.assertEqual(result, expectation)
def test_addition_and_multiplication_opcode(self):
prog = IntcodeComputer()
sample_op = [1, 1, 1, 4, 99, 5, 6, 0, 99]
expectation = [30, 1, 1, 4, 2, 5, 6, 0, 99]
result = prog.compute_sequence(sample_op)
self.assertEqual(result, expectation)
def test_multiplication_opcode(self):
prog = IntcodeComputer()
sample_op = [2, 3, 0, 3, 99]
expectation = [2, 3, 0, 6, 99]
result = prog.compute_sequence(sample_op)
self.assertEqual(result, expectation)
