profilarr/tests/regex.py

import sys
from roku import roku
from h265verify import h265
from qxr import qxr
from taoe import taoe
from ralphy import Ralphy
from av1 import AV1
# ... import other test functions

# ANSI escape codes for colors
BLUE = '\033[94m'
GREEN = '\033[92m'
RED = '\033[91m'
YELLOW = '\033[93m'
RESET = '\033[0m'

def run_tests():
    tests = [
        ("ROKU", roku),
        ("h265 Verified Groups", h265),
        ("QxR Groups", qxr),
        ("TAoE Groups", taoe),
        ("Ralphy", Ralphy),
        ("AV1", AV1),

        # ... add other test functions
    ]

    for test_name, test_func in tests:
        print(f"{BLUE}=============================================={RESET}")
        print(f"{BLUE}Running test: {test_name}{RESET}")
        print(f"{BLUE}=============================================={RESET}\n")
        
        test_result = test_func(debug_level=1)

        if test_result:
            print()
            print(f"{GREEN}=============================================={RESET}")
            print(f"{GREEN}Passed Test: {test_name}{RESET}")
            print(f"{GREEN}=============================================={RESET}\n")
            continue
        else:
            print()
            print(f"{RED}=============================================={RESET}")
            print(f"{RED}Failed Test: {test_name}{RESET}")
            print(f"{RED}=============================================={RESET}\n")
            sys.exit(1)

if __name__ == "__main__":
    run_tests()