with open('test_robustness_framework.py', 'r') as f:
    content = f.read()

# Fix the test to handle missing 'result' key
old_code = '''    print(f"Success: {result['success']}")
    print(f"Result: {result['result']}")
    print(f"Duration: {result['metadata']['performance']['duration_seconds']:.3f}s")'''

new_code = '''    print(f"Success: {result['success']}")
    if result['success']:
        print(f"Result: {result['result']}")
        print(f"Duration: {result['metadata']['performance']['duration_seconds']:.3f}s")
    else:
        print(f"Reason: {result.get('reason', 'Unknown error')}")
        print(f"Metadata: {result.get('metadata', {})}")'''

content = content.replace(old_code, new_code)

with open('test_robustness_framework.py', 'w') as f:
    f.write(content)

print("Fixed test script!")