def test_read_input_asks_input_generator():
computer = OpcodeComputer([])
computer.input_src = iter([0, 1, 2])
assert computer.read_input() == 0
assert computer.read_input() == 1
assert computer.read_input() == 2
def test_equals_example(input, expected):
computer = OpcodeComputer([3, 3, 1108, -1, 8, 3, 4, 3, 99])
computer.input_src = iter([input])
computer.output = Mock()
computer.run()
computer.output.assert_called_with(expected)
def test_lt_example(input, expected):
computer = OpcodeComputer(
[3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9])
computer.input_src = iter([input])
computer.output = Mock()
computer.run()
computer.output.assert_called_with(expected)
def test_large_example(input, expected):
computer = OpcodeComputer([
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_src = iter([input])
computer.output = Mock()
computer.run()
computer.output.assert_called_with(expected)
