def test_print_large_number():
program = [104, 1125899906842624, 99]
c = IntComputer()
def output(a):
assert a == 1125899906842624
def input():
return 8
c.output = output
c.input = input
result = c.run_program(program)
def test_immediate_mode_input_less_than_8():
program = [3, 9, 7, 9, 10, 9, 4, 9, 99, -1, 8]
c = IntComputer()
def output(a):
assert a == 0
def input():
return 8
c.output = output
c.input = input
result = c.run_program(program)
assert result == [3, 9, 7, 9, 10, 9, 4, 9, 99, 0, 8]
def test_immediate_mode_input_equals_8():
program = [3, 9, 8, 9, 10, 9, 4, 9, 99, -1, 8]
c = IntComputer()
def output(a):
assert a == 1
def input():
return 8
c.output = output
c.input = input
result = c.run_program(program)
assert result == [3, 9, 8, 9, 10, 9, 4, 9, 99, 1, 8]
