Run all test cases for a college assignment, usually provided as .in and .out files.

assignment_tests.py

import glob
import sys
import io

files = glob.glob('./*.in')
for f in files:
    test_case = f.split('.in')[0].replace('.\\', '')
    sys.stdin = open(f, "r")

    print(f"\n*** Caso de Teste {test_case} ***\n")

    stdout = sys.stdout
    output = io.StringIO()
    sys.stdout = output

    main()

    output = output.getvalue()
    with open(f"{test_case}.out", "r") as fp:
        expected_output = fp.read()

    sys.stdout = stdout
    output = output.splitlines()
    expected_output = expected_output.splitlines()

    for i, (output_line, expected_output_line) in enumerate(zip(output, expected_output)):
        if (output_line == expected_output_line):
            valid = "🟢"
        else:
            valid = "🔴"

        print(f"Caso {i+1}\tSaída: {output_line}\tSaída Esperada: {expected_output_line}\t{valid}")
print("\n")