File size: 2,035 Bytes
77da5ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import sys
import os
sys.path.append(os.getcwd())
from core.reward import compute_task_reward
from core.life_state import LifeMetrics
from core.task import Task
def test_reasoning_alignment_pass():
"""Verify that reasoning mentioning the action category gets higher score."""
state = LifeMetrics()
task = Task(
"test",
"career",
"test",
{}, {}, {}, {}, # constraints, hidden, mutable, visible
[], [], # success, failure
[], [], [], # events, routes, milestones
10, 1, {} # horizon, diff, metadata
)
# Action: SPEND, Reasoning: mentions cost
_, res_match = compute_task_reward(
state_before=state, state_after=state, resources_used={}, actions_taken=1,
milestones_achieved=[], success_conditions_met=[], exo_events_seen=0,
milestones_after_event=0, routes_remaining=1, rollback_used=False,
cascade_collapse=False, task=task, action_type="spend",
reasoning="I am doing this because the cost is low."
)
# Action: SPEND, Reasoning: mentions nothing relevant
_, res_mismatch = compute_task_reward(
state_before=state, state_after=state, resources_used={}, actions_taken=1,
milestones_achieved=[], success_conditions_met=[], exo_events_seen=0,
milestones_after_event=0, routes_remaining=1, rollback_used=False,
cascade_collapse=False, task=task, action_type="spend",
reasoning="I am doing this because I am happy."
)
score_match = res_match["components"]["reasoning"]
score_mismatch = res_mismatch["components"]["reasoning"]
print(f"Match Score: {score_match}")
print(f"Mismatch Score: {score_mismatch}")
assert score_match > score_mismatch, f"Match {score_match} should be > Mismatch {score_mismatch}"
print("✅ Reasoning alignment test passed!")
if __name__ == "__main__":
try:
test_reasoning_alignment_pass()
except Exception as e:
print(f"❌ Test failed: {e}")
sys.exit(1)
|