1"""Tests for data models."""
2 
3from datetime import datetime
4 
5from reasoning_trace_optimizer.models import (
6    AnalysisResult,
7    LoopResult,
8    OptimizationResult,
9    Pattern,
10    PatternType,
11    PromptDiff,
12    ReasoningTrace,
13    Severity,
14    ThinkingBlock,
15    ToolCall,
16)
17 
18 
19def test_thinking_block_creation():
20    """Test ThinkingBlock creation with defaults."""
21    block = ThinkingBlock(
22        content="This is a test thinking block.",
23        turn_index=0,
24    )
25    assert block.content == "This is a test thinking block."
26    assert block.turn_index == 0
27    assert block.token_count == 0
28    assert isinstance(block.timestamp, datetime)
29 
30 
31def test_tool_call_creation():
32    """Test ToolCall creation."""
33    tc = ToolCall(
34        id="call_123",
35        name="get_weather",
36        input={"location": "San Francisco"},
37        turn_index=1,
38    )
39    assert tc.id == "call_123"
40    assert tc.name == "get_weather"
41    assert tc.input["location"] == "San Francisco"
42    assert tc.success is None
43 
44 
45def test_reasoning_trace_creation():
46    """Test ReasoningTrace creation and methods."""
47    trace = ReasoningTrace(
48        session_id="test-session",
49        task="Test task",
50        system_prompt="Test prompt",
51    )
52 
53    # Add thinking block
54    block = ThinkingBlock(content="Thinking...", turn_index=0)
55    trace.thinking_blocks.append(block)
56 
57    # Add tool call
58    tc = ToolCall(
59        id="call_1",
60        name="test_tool",
61        input={},
62        turn_index=0,
63    )
64    trace.tool_calls.append(tc)
65 
66    # Test methods
67    assert trace.get_thinking_at_turn(0) == block
68    assert trace.get_thinking_at_turn(1) is None
69    assert len(trace.get_tool_calls_at_turn(0)) == 1
70    assert len(trace.get_tool_calls_at_turn(1)) == 0
71 
72 
73def test_pattern_creation():
74    """Test Pattern creation."""
75    pattern = Pattern(
76        type=PatternType.CONTEXT_DEGRADATION,
77        severity=Severity.HIGH,
78        description="Model lost track of goal",
79        evidence=["Evidence 1", "Evidence 2"],
80        turn_indices=[2, 3],
81        suggestion="Add explicit reminders",
82        confidence=0.85,
83    )
84    assert pattern.type == PatternType.CONTEXT_DEGRADATION
85    assert pattern.severity == Severity.HIGH
86    assert pattern.confidence == 0.85
87 
88 
89def test_analysis_result_creation():
90    """Test AnalysisResult creation."""
91    result = AnalysisResult(trace_id="test-trace")
92    assert result.overall_score == 0.0
93    assert len(result.patterns) == 0
94    assert len(result.recommendations) == 0
95 
96 
97def test_optimization_result_creation():
98    """Test OptimizationResult creation."""
99    result = OptimizationResult(
100        original_prompt="Original",
101        optimized_prompt="Optimized",
102    )
103    result.diffs.append(PromptDiff(
104        section="instructions",
105        original="Original text",
106        optimized="Improved text",
107        reason="Better clarity",
108    ))
109    assert len(result.diffs) == 1
110    assert result.diffs[0].section == "instructions"
111 
112 
113def test_loop_result_creation():
114    """Test LoopResult creation."""
115    result = LoopResult(task="Test task")
116    assert result.total_iterations == 0
117    assert result.converged is False
118    assert result.improvement_percentage == 0.0
119 
120 
121def test_pattern_types():
122    """Test all PatternType values exist."""
123    expected_types = [
124        "context_degradation",
125        "tool_confusion",
126        "instruction_drift",
127        "hallucination",
128        "incomplete_reasoning",
129        "tool_misuse",
130        "goal_abandonment",
131        "circular_reasoning",
132        "premature_conclusion",
133        "missing_validation",
134    ]
135    for type_name in expected_types:
136        assert PatternType(type_name) is not None
137 
138 
139def test_severity_levels():
140    """Test all Severity levels exist."""
141    assert Severity.LOW.value == "low"
142    assert Severity.MEDIUM.value == "medium"
143    assert Severity.HIGH.value == "high"
144    assert Severity.CRITICAL.value == "critical"
145