95 lines
3.5 KiB
Python
95 lines
3.5 KiB
Python
|
"""Tests for Consciousness Engineering — formal self-model."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fusionagi.reasoning.self_model import (
|
||
|
|
AttentionFocus,
|
||
|
|
CognitiveState,
|
||
|
|
SelfModel,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestSelfModel:
|
||
|
|
def test_initial_state(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
assert sm.cognitive_state == CognitiveState.IDLE
|
||
|
|
assert sm.attention_focus == AttentionFocus.TASK
|
||
|
|
|
||
|
|
def test_set_state(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.set_state(CognitiveState.REASONING, AttentionFocus.INTERNAL_STATE, "thinking hard")
|
||
|
|
assert sm.cognitive_state == CognitiveState.REASONING
|
||
|
|
assert sm.attention_focus == AttentionFocus.INTERNAL_STATE
|
||
|
|
|
||
|
|
def test_register_and_update_capability(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.register_capability("logic", "formal reasoning", initial_confidence=0.6)
|
||
|
|
sm.update_capability("logic", success=True)
|
||
|
|
sm.update_capability("logic", success=True)
|
||
|
|
sm.update_capability("logic", success=False)
|
||
|
|
report = sm.introspect()
|
||
|
|
assert "logic" in report["capabilities"]
|
||
|
|
assert report["capabilities"]["logic"]["evidence_count"] == 3
|
||
|
|
|
||
|
|
def test_goal_management(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.set_goal("g1", "Learn from mistakes", priority=0.9)
|
||
|
|
sm.update_goal_progress("g1", 0.5)
|
||
|
|
report = sm.introspect()
|
||
|
|
assert "g1" in report["goals"]
|
||
|
|
assert report["goals"]["g1"]["progress"] == 0.5
|
||
|
|
|
||
|
|
def test_goal_alignment_check(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.set_goal("g1", "Test goal")
|
||
|
|
sm._goals["g1"].aligned_with_values = False
|
||
|
|
warnings = sm.check_goal_alignment()
|
||
|
|
assert any("conflict" in w for w in warnings)
|
||
|
|
|
||
|
|
def test_emotional_state_update(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.update_emotional_state("confidence", 0.3)
|
||
|
|
report = sm.introspect()
|
||
|
|
assert report["emotional_state"]["confidence"] > 0.5
|
||
|
|
|
||
|
|
def test_emotional_state_clamped(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.update_emotional_state("confidence", 10.0)
|
||
|
|
assert sm._emotional_state["confidence"] == 1.0
|
||
|
|
sm.update_emotional_state("confidence", -20.0)
|
||
|
|
assert sm._emotional_state["confidence"] == 0.0
|
||
|
|
|
||
|
|
def test_explain_state(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.set_state(CognitiveState.REASONING, AttentionFocus.TASK)
|
||
|
|
explanation = sm.explain_state()
|
||
|
|
assert "reasoning" in explanation
|
||
|
|
assert "task" in explanation
|
||
|
|
|
||
|
|
def test_introspect_returns_all_fields(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
report = sm.introspect()
|
||
|
|
assert "cognitive_state" in report
|
||
|
|
assert "attention_focus" in report
|
||
|
|
assert "capabilities" in report
|
||
|
|
assert "goals" in report
|
||
|
|
assert "values" in report
|
||
|
|
assert "emotional_state" in report
|
||
|
|
assert "recent_thoughts" in report
|
||
|
|
|
||
|
|
def test_introspection_log_trimming(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm._max_log_size = 10
|
||
|
|
for i in range(200):
|
||
|
|
sm.set_state(CognitiveState.REASONING, thought=f"thought_{i}")
|
||
|
|
# After exceeding max_log_size, the log is trimmed to notable + last 100
|
||
|
|
assert len(sm._introspection_log) <= 120
|
||
|
|
|
||
|
|
def test_get_summary(self) -> None:
|
||
|
|
sm = SelfModel()
|
||
|
|
sm.register_capability("test", "test cap")
|
||
|
|
sm.set_goal("g1", "test goal")
|
||
|
|
summary = sm.get_summary()
|
||
|
|
assert summary["capabilities_count"] == 1
|
||
|
|
assert summary["goals_count"] == 1
|