stress_testing.py

from helpers import Colors
from utils import cpp_compiler
from utils import file_handler


def stress_testing():
    test_cases = int(
        input(f"{Colors.fg.gold}Enter the number of test cases: {Colors.reset}")
    )
    time_limit = float(
        input(
            f"{Colors.fg.gold}Enter the time limit for each test case (seconds): {Colors.reset}"
        )
    )

    print(f"{Colors.fg.white}\n=========================={Colors.reset}\n")

    list_of_files = ["generator", "correct", "test"]
    list_of_input_files = ["", "input.txt", "input.txt"]
    list_of_output_files = ["input.txt", "correct_output.txt", "test_output.txt"]

    for file in list_of_files:
        try:
            cpp_compiler.compile_file(file_name=file)
        except ValueError as e:
            print(f"\n{Colors.fg.red}{e}{Colors.reset}")
            return

    for test in range(1, test_cases + 1):
        for index in range(len(list_of_files)):
            try:
                cpp_compiler.validate_errors(
                    list_of_files[index],
                    list_of_input_files[index],
                    list_of_output_files[index],
                    test,
                    time_limit,
                )
            except Exception as e:
                print(f"\n{Colors.fg.red}{e}{Colors.reset}")
                return

        if cpp_compiler.check_diff("correct_output.txt", "test_output.txt"):
            print(
                f"{Colors.fg.white}Test case {test} {Colors.fg.green}passed! ✅{Colors.reset}"
            )
        else:
            print(
                f"{Colors.fg.red}Wrong Answer {Colors.fg.white}on test case {test} ❌{Colors.reset}"
            )

            print(f"{Colors.fg.white}\n=========================={Colors.reset}\n")
                
            print(f"{Colors.fg.purple}Input: {Colors.reset}")
            cpp_compiler.print_output("input.txt")

            print(f"{Colors.fg.purple}Expected Output: {Colors.reset}")
            cpp_compiler.print_output("correct_output.txt")

            print(f"{Colors.fg.purple}Your Output: {Colors.reset}")
            cpp_compiler.print_output("test_output.txt")

            print(f"{Colors.fg.white}=========================={Colors.reset}\n")

            return

    print(f"{Colors.fg.white}==========================\n{Colors.reset}")

    print(f"\n{Colors.fg.pink}All test cases {Colors.fg.green}passed! ✅{Colors.reset}")


if __name__ == "__main__":
    stress_testing()
    file_handler.clean_up()