def test_offset_1(): program = [109, 19, 99] c = IntComputer() c.offset = 2000 c.output = Mock() result = c.run_program(program) assert c.offset == 2019
def test_offset_2(): program = [109, 19, 204, -34, 99] c = IntComputer() c.offset = 2000 c.output = Mock() result = c.run_program(program) assert c.offset == 2019 assert c.output.call_count == 1