#!/usr/bin/env python3
"""
Test script to verify COB RL model extraction
"""

print('=== COB RL MODEL AUDIT ===')

# Test the extracted COB RL model
try:
    from NN.models.cob_rl_model import MassiveRLNetwork, COBRLModelInterface
    
    print('1. Testing MassiveRLNetwork...')
    model = MassiveRLNetwork()
    model_info = model.get_model_info()
    
    print('   ✓ Model created successfully')
    print(f'   ✓ Total parameters: {model_info["total_parameters"]:,}')
    print(f'   ✓ Architecture: {model_info["architecture"]}')
    
    print('2. Testing COBRLModelInterface...')
    interface = COBRLModelInterface()
    print(f'   ✓ Interface created on device: {interface.device}')
    
    print('3. Testing imports from models module...')
    from NN.models import MassiveRLNetwork as ImportedModel
    print('   ✓ Models properly exported')
    
    print()
    print('=== AUDIT SUMMARY ===')
    print('✅ COB RL MODEL EXTRACTION SUCCESSFUL')
    print('🧠 The 1B parameter model is now in: NN/models/cob_rl_model.py')
    print('   - MassiveRLNetwork: Core transformer model')
    print('   - COBRLModelInterface: Training/inference interface')
    print()
    print('📊 Model Details:')
    print(f'   - Parameters: {model_info["total_parameters"]:,} (~1B)')
    print(f'   - Input size: {model_info["input_size"]} (COB features)')
    print(f'   - Hidden size: {model_info["hidden_size"]}')
    print(f'   - Layers: {model_info["num_layers"]}')
    print('   - Output: Price direction, value, confidence')
    
except Exception as e:
    print(f'❌ Audit failed: {e}')
    import traceback
    traceback.print_exc()