stess_test.py

from random import randint, seed
from os import system

seed(9)

CORRECT_FILE = "OK.cpp"
INCORRECT_FILE = "WA.cpp"
TEST_CASE_FILE = "testcase.txt"

system(f"g++ {CORRECT_FILE} -o CORRECT")
system(f"g++ {INCORRECT_FILE} -o INCORRECT")


def write_test(input):
    open(TEST_CASE_FILE, "w").write(input)


def generate():
    t = randint(*t_range)
    s = f"{t}\n"
    for _ in range(t):
        s += f"{randint(*int_range)} {randint(*int_range)}\n"
    return s


def get_ans(solution_name):
    system(f"./{solution_name} < {TEST_CASE_FILE} > out.txt")
    return open("out.txt", "r").read()


int_range = (1, 337)
t_range = (1, 2)
test_cases = 2
for i in range(test_cases):
    testcase = generate()
    write_test(testcase)

    CORRECT_ANS = get_ans("./CORRECT")
    INCORRECT_ANS = get_ans("./INCORRECT")
    if CORRECT_ANS != INCORRECT_ANS:
        print("test_case", i + 1, "\033[91mWA\033[0m")
        print("\33[34mInput:\033[0m")
        print(testcase)
        print("\33[34mIncorrect Output:\033[0m")
        print(INCORRECT_ANS)
        print("\33[34mExpected:\033[0m")
        print(CORRECT_ANS)
    else:
        print("test_case", i + 1, "\33[32mOK\033[0m")