test_comprehensive_pipeline.py

#!/usr/bin/env python3
"""
Test script for comprehensive pipeline with qualia feelings integration
"""

from src.pipelines.comprehensive_pipeline import BIOMINDComprehensivePipeline
import torch

def test_comprehensive_pipeline():
    """Test the comprehensive pipeline with qualia feelings"""

    # Initialize pipeline
    pipeline = BIOMINDComprehensivePipeline(device='cpu')

    # Test question
    question = 'What is the capital of France?'
    choices = ['London', 'Paris', 'Berlin', 'Rome']

    print('[BRAIN] Testing comprehensive pipeline with qualia feelings integration...')

    # Process question
    result = pipeline.process_question_comprehensive(
        question=question,
        choices=choices,
        expected_answer=1,  # Paris is correct
        enable_debug_logging=True
    )

    print(f'\n[OK] Result: {result["success"]}')
    print(f'[TARGET] Answer: {result["pipeline_results"]["final_synthesis"]["extracted_answer"]}')

    # Check qualia states
    if result['qualia_states']:
        print(f'\n? Qualia states captured: {len(result["qualia_states"])}')

        # Show final qualia state
        final_qualia = result['qualia_states'][-1]['qualia_metrics']
        print(f'Final qualia feelings: {final_qualia["feelings"]}')

    return result

if __name__ == '__main__':
    test_comprehensive_pipeline()