Тестирование .py файлов

import io
import sys
 
old_stdout, sys.stdout = sys.stdout, io.StringIO()
 
files = ["1.py", "1.py", "2.py", "3.py"]
 
for file in files:
    try:
        with open(file, "rt") as f_handle:
            exec(f_handle.read())
        if sys.stdout.getvalue() == "1 2 3\n":
            print(f"File: {file} is accept!", file=old_stdout)
        else:
            print(f"File: {file} wrong answer!", file=old_stdout)
        sys.stdout.seek(0)
    except Exception as error:
        print(error, file=old_stdout)
 
# File: 1.py is accept!
# File: 1.py is accept!
# [Errno 2] No such file or directory: '2.py'
# File: 3.py wrong answer!

print(3 * 3)

x = 3
result = x ** 2
print(result)

print(3**3)

import os
import subprocess
 
 
py_files = os.listdir('code')
 
with open(os.path.join('tasks', '1.a.txt')) as f:
    answer = f.read()
 
for py_file in py_files:
    file = os.path.abspath(os.path.join('code', py_file))
 
    p = subprocess.Popen(f'python {file}', stdout=subprocess.PIPE, shell=True)
    out, err = p.communicate()
    print(out.decode().strip() == answer)

print(int(input()) ** 2)

a = int(input())
b = int(input())...

a = int(input())...

for _ in range(int(input()):
a = int(input())...

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

from checking import *
 
 
@test
def check_task():
    expected_inputs = ['3', '3']  # что введут в инпуты (мы это достаем из файла)
    expected_result = 27  # что должно быть на выходе (достаем из второго файла)
    iterator = iter(expected_inputs)  # приходится взять итератор, чтобы поэлементно доставать инпуты (по очереди, имитируя 2 ввода)
    print_list = []  # сюда закинем вывод принта
    with mock_builtins('print', lambda x: print_list.append(x)):  # подменяем реальный принт на добавление в список
        with mock_builtins('input', lambda: next(iterator)):  # подменяем реальный инпут на запрос следующего значения из списка инпутов
            import main.some_module  # импортируем модуль ученика, весь код в нем выполняется, но принты и инпуты уже наши, тестовые
 
    equals(expected_result, print_list[0])  # убеждаемся что результат по типу и значению совпадает с ожидаемым
 
 
if __name__ == '__main__':
    start(3)  # запускаем тест

Starting suite "Default Test Suite"
----------
Test "__main__.check_task"  SUCCESS!
 
==============================
Total tests: 1, success tests: 1, failed tests: 0, broken tests: 0, ignored tests: 0
Time elapsed: 0.00 seconds.
==============================