run_tests.py

import subprocess
import re
import os
import sys

# Configuration
SIMULATOR_BIN = "simulator.exe" if os.name == 'nt' else "./simulator"
SOURCE_FILES = "src/main.cpp src/core/SparseFlowCore.cpp src/hardware/RegisterFile.cpp src/hardware/Memory.cpp src/hardware/BranchPredictor.cpp src/hardware/Cache.cpp src/utils/Loader.cpp src/utils/Stats.cpp src/utils/TraceLogger.cpp"
COMPILE_CMD = f"g++ -o {SIMULATOR_BIN} {SOURCE_FILES}"

TESTS = [
    {
        "name": "Hazard Detection (Load-Use)",
        "file": "tests/test_hazard.asm",
        "expected_regs": {"t2": 109}, # 99 + 10
        # Stall detection verified by correct register value (no debug output in production)
    },
    {
        "name": "Forwarding (EX-EX)",
        "file": "tests/test_forwarding.asm",
        "expected_regs": {"t2": 30},
        # Forwarding verified by correct result without stalls
    },
    {
        "name": "Branch Prediction (Taken)",
        "file": "tests/test_branch.asm",
        "expected_regs": {"t2": 777},
        # Branch handling verified by correct target execution
    }
]

def compile_simulator():
    print(f"Compiling: {COMPILE_CMD}")
    result = subprocess.run(COMPILE_CMD, shell=True, capture_output=True, text=True)
    if result.returncode != 0:
        print("Compilation Failed!")
        print(result.stderr)
        return False
    return True

def parse_registers(output):
    regs = {}
    # Regex to match "x5(t0)    : 10"
    pattern = re.compile(r"x\d+\(([\w\d]+)\)\s+:\s+(\d+)")
    for line in output.splitlines():
        match = pattern.search(line)
        if match:
            name = match.group(1)
            val = int(match.group(2))
            regs[name] = val
    return regs

def run_test(test):
    print(f"Running Test: {test['name']}...")
    
    cmd = f"{SIMULATOR_BIN} {test['file']}"
    # Use shell=True for Windows to handle paths correctly
    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
    
    if result.returncode != 0:
        print(f"FAIL: Simulator crashed (Exit {result.returncode})")
        return False

    output = result.stdout
    regs = parse_registers(output)
    
    passed = True
    
    # Check Registers
    if "expected_regs" in test:
        for reg, val in test["expected_regs"].items():
            if reg not in regs:
                print(f"  FAIL: Register {reg} not found in output.")
                passed = False
            elif regs[reg] != val:
                print(f"  FAIL: Register {reg} expected {val}, got {regs[reg]}")
                passed = False
                
    # Check Output Strings
    if "expected_output" in test:
        for s in test["expected_output"]:
            if s not in output:
                print(f"  FAIL: Expected output string '{s}' not found.")
                passed = False

    if "must_not_contain" in test:
        for s in test["must_not_contain"]:
            if s in output:
                print(f"  FAIL: Output contained forbidden string '{s}'.")
                passed = False

    if passed:
        print("  PASS")
    return passed

def main():
    if not compile_simulator():
        sys.exit(1)
    
    total = len(TESTS)
    passed = 0
    
    for test in TESTS:
        if run_test(test):
            passed += 1
            
    print(f"\nTest Summary: {passed}/{total} Passed.")
    if passed != total:
        sys.exit(1)

if __name__ == "__main__":
    main()