На linux, я могу использовать pexpect, и я пишу такой скрипт:
# coding:utf8
# python2
'''
input 9: 9
input 1: 1
Right!
'''
import pexpect
p = pexpect.spawn('python simple_check.py')
p.expect(': ')
first_num = p.before.split(' ')[1]
p.send(first_num+'\n')
p.expect(first_num)
p.expect(': ')
second_num = p.before.split(' ')[1]
p.sendline(second_num)
p.interact()
Вывод:
4
Right!
По крайней мере, он работает на linux. Я постараюсь найти способ на windows.