def part1(input_file="input/day_07"):
results = []
perm = permutations([0, 1, 2, 3, 4])
for p in perm:
amp_a = IntCode()
amp_a.get_input(input_file)
amp_a.set_user_input([p[0], 0])
amp_a.diagnostic_program()
result_a = amp_a.result
amp_b = IntCode()
amp_b.get_input(input_file)
amp_b.set_user_input([p[1], result_a])
amp_b.diagnostic_program()
result_b = amp_b.result
amp_c = IntCode()
amp_c.get_input(input_file)
amp_c.set_user_input([p[2], result_b])
amp_c.diagnostic_program()
result_c = amp_c.result
amp_d = IntCode()
amp_d.get_input(input_file)
amp_d.set_user_input([p[3], result_c])
amp_d.diagnostic_program()
result_d = amp_d.result
amp_e = IntCode()
amp_e.get_input(input_file)
amp_e.set_user_input([p[4], result_d])
amp_e.diagnostic_program()
result_e = amp_e.result
if result_e == 298586:
print(p)
results.append(result_e)
print(max(results))
return max(results)
def test_halt_and_restart():
intcode = IntCode()
intcode.get_input("input/day_07")
intcode.set_user_input([1, 2])
intcode.diagnostic_program()
return_value = intcode.result
intcode = IntCode()
intcode.get_input("input/day_07")
intcode.set_user_input([1])
intcode.diagnostic_program()
intcode.set_user_input([2])
intcode.diagnostic_program()
assert return_value == intcode.result
def test_d2(input_file, user_input, expected_result):
intcode = IntCode()
intcode.get_input(input_file)
intcode.set_user_input(user_input)
assert intcode.diagnostic_program() == expected_result
def test_part1():
intcode = IntCode()
intcode.get_input("input/day_05")
intcode.set_user_input([1])
intcode.diagnostic_program()
assert intcode.result == 16489636
def part2(input_file="input/day_07"):
perm = permutations([5, 6, 7, 8, 9])
results = []
for p in perm:
phase_seq = str(p[0]) + str(p[1]) + str(p[2]) + str(p[3]) + str(p[4])
result_e = 0
amp_a = IntCode()
amp_a.get_input(input_file)
amp_a.set_user_input([p[0]])
amp_b = IntCode()
amp_b.get_input(input_file)
amp_b.set_user_input([p[1]])
amp_c = IntCode()
amp_c.get_input(input_file)
amp_c.set_user_input([p[2]])
amp_d = IntCode()
amp_d.get_input(input_file)
amp_d.set_user_input([p[3]])
amp_e = IntCode()
amp_e.get_input(input_file)
amp_e.set_user_input([p[4]])
while True:
amp_a.set_user_input([result_e])
amp_a.diagnostic_program()
result_a = amp_a.result
amp_b.set_user_input([result_a])
amp_b.diagnostic_program()
result_b = amp_b.result
amp_c.set_user_input([result_b])
amp_c.diagnostic_program()
result_c = amp_c.result
amp_d.set_user_input([result_c])
amp_d.diagnostic_program()
result_d = amp_d.result
amp_e.set_user_input([result_d])
amp_e.diagnostic_program()
result_e = amp_e.result
if result_e == 9246095:
print(p)
if amp_e.halt_code_reached:
# print(phase_seq, result_e)
results.append(result_e)
break
print(max(results))
return max(results)