File size: 1,002 Bytes
7bf0480
adea8c3
 
7bf0480
 
adea8c3
 
7bf0480
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import sys
from codelens_env.models import TaskId
from codelens_env.env import CodeLensEnv

def validate_environment():
    print("=== CodeLens Validation Suite ===")
    env = CodeLensEnv()
    
    tasks = [TaskId.BUG_DETECTION, TaskId.SECURITY_AUDIT, TaskId.ARCHITECTURAL_REVIEW]
    total_scenarios = 30
    passed = 0
    
    for task in tasks:
        print(f"\nValidating Task: {task}")
        for seed in range(10): # Check first 10 scenarios per task
            try:
                env.reset(task, seed=seed)
                passed += 1
                print(f"  [PASS] Scenario {seed}")
            except Exception as e:
                print(f"  [FAIL] Scenario {seed}: {e}")
                
    print(f"\nFinal Result: {passed}/{total_scenarios} scenarios reachable.")
    if passed == 30:
        print("ENVIRONMENT VALIDATED SUCCESSFULLY.")
        sys.exit(0)
    else:
        print("VALIDATION FAILED.")
        sys.exit(1)

if __name__ == "__main__":
    validate_environment()