Как написать "тестирующую" систему (pyhton)

Допустим я получаю файл содержания:

a = int(input())
b = int(input())
print(a + b)

Далее мне нужно проверить верна ли выведенная в консоль информация. Я бы делал это через assert, но я не понимаю, как мне получить результат работы программы.

Каким образом это можно реализовать?