"""Tests für data_layer.prompt_output_compact (KI-Platzhalter, Token)."""

import pytest

from data_layer.prompt_output_compact import (
    compact_float_for_prompt,
    compact_json_payload_for_prompts,
    format_scalar_for_prompt_text,
    normalize_prompt_number,
    session_metrics_list_to_key_value_compact,
)


@pytest.mark.parametrize(
    "x,expected",
    [
        (0.0, 0),
        (123.456, 123),
        (45.67, 46),
        (9.876, 9.88),
        (0.99, 0.99),
        (0.055, 0.055),
        (0.01234, 0.012),
    ],
)
def test_compact_float_for_prompt(x, expected):
    out = compact_float_for_prompt(x)
    if isinstance(expected, float):
        assert abs(float(out) - expected) < 0.0001
    else:
        assert out == expected


def test_compact_json_nested():
    raw = {"a": 12.345678, "b": {"c": 0.0666}, "d": [1.111, 2.0]}
    out = compact_json_payload_for_prompts(raw)
    assert out["a"] == 12
    assert abs(out["b"]["c"] - 0.067) < 0.001
    assert out["d"][0] == 1.11


def test_format_scalar_no_long_float_tail():
    s = format_scalar_for_prompt_text(51.5818181818181818)
    assert "181818" not in s
    assert len(s) <= 8


def test_format_scalar_numeric_string_no_long_tail():
    s = format_scalar_for_prompt_text("51.581818181818181818")
    assert "181818" not in s


def test_session_metrics_string_dtype_compacts_numeric_strings():
    sm = [
        {
            "key": "temp_c",
            "data_type": "string",
            "value": "22.333333333333336",
        },
        {
            "key": "kcal_per_km",
            "data_type": "string",
            "value": "51.581818181818181818",
        },
    ]
    out = session_metrics_list_to_key_value_compact(sm)
    assert out["temp_c"] == 22
    assert out["kcal_per_km"] == 52


def test_session_metrics_key_value_only():
    sm = [
        {
            "key": "rpe",
            "data_type": "integer",
            "value": 7,
            "name_de": "RPE",
            "description_de": "lang",
        },
        {
            "key": "watts",
            "data_type": "float",
            "value": 199.999,
            "unit": "W",
        },
    ]
    out = session_metrics_list_to_key_value_compact(sm)
    assert out == {"rpe": 7, "watts": 200}
    assert "name_de" not in str(out)