app/test_duration_align.py

"""Specialty-aware duration steering tests."""
from __future__ import annotations

from .duration_align import align_time_spent, duration_hint_for_llm


def test_fp_brief_time_steers_to_minor_not_paeds():
    align = align_time_spent(8, specialty_code="00")
    assert align is not None
    assert "A001" in align.keywords
    assert "paediatric" not in align.keywords.lower()


def test_fp_duration_hint_mentions_office_ladder():
    align = align_time_spent(10, specialty_code="00")
    hint = duration_hint_for_llm(align, specialty_code="00")
    assert "A001" in hint
    assert "A260" not in hint


def test_paeds_brief_time_still_steers_level_1():
    align = align_time_spent(12, specialty_code="26")
    assert align is not None
    assert "level 1" in align.keywords.lower()