"""Unit tests for INCUBAZPRIMARIA model_value rounding and labels."""

from __future__ import annotations

import unittest
from decimal import Decimal

from pipeline.stages.disease import (
    _DayRecords,
    _format_model_value,
    _incub_label,
    _resolve_day_fase,
)


class FormatModelValueTests(unittest.TestCase):
    def test_long_float_rounds_to_two_decimals(self) -> None:
        self.assertEqual(_format_model_value(40.123456789), "40.12")

    def test_long_decimal_rounds_to_two_decimals(self) -> None:
        self.assertEqual(_format_model_value(Decimal("40.123456789")), "40.12")

    def test_trailing_zeros_stripped(self) -> None:
        self.assertEqual(_format_model_value(40.30), "40.3")
        self.assertEqual(_format_model_value(40.0), "40")
        self.assertEqual(_format_model_value(Decimal("40.10")), "40.1")

    def test_none_and_empty_return_none(self) -> None:
        self.assertIsNone(_format_model_value(None))
        self.assertIsNone(_format_model_value("  "))

    def test_non_numeric_passthrough(self) -> None:
        self.assertEqual(_format_model_value("n/a"), "n/a")


class IncubLabelTests(unittest.TestCase):
    def test_picks_max_raw_value_then_rounds_suffix(self) -> None:
        self.assertEqual(
            _incub_label([
                ("INCUBAZPRIMARIA", 40.12),
                ("INCUBAZPRIMARIA", 40.129),
                ("INCUBAZPRIMARIA", 39.99),
            ]),
            "INCUBAZPRIMARIA_40.13",
        )

    def test_keeps_index_of_winning_row(self) -> None:
        self.assertEqual(
            _incub_label([("INCUBAZPRIMARIA2", 80.7)]),
            "INCUBAZPRIMARIA2_80.7",
        )

    def test_higher_percentage_beats_higher_index(self) -> None:
        self.assertEqual(
            _incub_label([
                ("INCUBAZPRIMARIA2", 40),
                ("INCUBAZPRIMARIA1", 90),
            ]),
            "INCUBAZPRIMARIA1_90",
        )

    def test_tie_keeps_higher_index(self) -> None:
        self.assertEqual(
            _incub_label([
                ("INCUBAZPRIMARIA1", 80.7),
                ("INCUBAZPRIMARIA2", 80.7),
            ]),
            "INCUBAZPRIMARIA2_80.7",
        )

    def test_missing_index_loses_tie(self) -> None:
        self.assertEqual(
            _incub_label([
                ("INCUBAZPRIMARIA", 40),
                ("INCUBAZPRIMARIA2", 40),
            ]),
            "INCUBAZPRIMARIA2_40",
        )

    def test_space_or_underscore_is_not_an_index(self) -> None:
        self.assertEqual(
            _incub_label([("INCUBAZPRIMARIA 2", 40), ("INCUBAZPRIMARIA_2", 30)]),
            "INCUBAZPRIMARIA_40",
        )

    def test_empty_or_non_numeric_falls_back(self) -> None:
        self.assertEqual(_incub_label([]), "INCUBAZPRIMARIA")
        self.assertEqual(
            _incub_label([("INCUBAZPRIMARIA", None), ("INCUBAZPRIMARIA2", "n/a")]),
            "INCUBAZPRIMARIA",
        )

    def test_day_keeps_each_index_and_fase5(self) -> None:
        records = _DayRecords(
            descriptions={"INCUBAZPRIMARIA1", "INCUBAZPRIMARIA2", "FASE5", "FASE2"},
            incub_rows=[
                ("INCUBAZPRIMARIA1", 90),
                ("INCUBAZPRIMARIA2", 40),
            ],
        )
        self.assertEqual(
            _resolve_day_fase(records),
            ["INCUBAZPRIMARIA1_90", "INCUBAZPRIMARIA2_40", "FASE5"],
        )

    def test_incubation_only_keeps_highest_value_per_index(self) -> None:
        records = _DayRecords(
            descriptions={"INCUBAZPRIMARIA2"},
            incub_rows=[
                ("INCUBAZPRIMARIA2", 40),
                ("INCUBAZPRIMARIA2", 80.7),
            ],
        )
        self.assertEqual(_resolve_day_fase(records), ["INCUBAZPRIMARIA2_80.7"])

    def test_fase2_without_incubation(self) -> None:
        records = _DayRecords(descriptions={"FASE2"})
        self.assertEqual(_resolve_day_fase(records), ["FASE2"])

    def test_empty_day_is_none(self) -> None:
        self.assertIsNone(_resolve_day_fase(_DayRecords()))


if __name__ == "__main__":
    unittest.main()