"""Unit tests for analysis.data_audit.""" from __future__ import annotations from datetime import UTC, datetime from cerbero_bite.analysis.data_audit import _expected_ticks # noqa: PLC2701 def test_expected_ticks_basic() -> None: since = datetime(2026, 5, 12, 12, 0, tzinfo=UTC) until = datetime(2026, 5, 12, 13, 0, tzinfo=UTC) # 12:00, 12:15, 12:30, 12:45 → 4 ticks before 13:00 assert _expected_ticks(since, until) == 4 def test_expected_ticks_inclusive_left_exclusive_right() -> None: since = datetime(2026, 5, 12, 12, 0, tzinfo=UTC) until = datetime(2026, 5, 12, 12, 15, tzinfo=UTC) # Only 12:00 assert _expected_ticks(since, until) == 1 def test_expected_ticks_unaligned_since_rounds_up() -> None: since = datetime(2026, 5, 12, 12, 7, tzinfo=UTC) until = datetime(2026, 5, 12, 12, 30, tzinfo=UTC) # First aligned tick after 12:07 is 12:15, then 12:30 is excluded assert _expected_ticks(since, until) == 1 def test_expected_ticks_non_multiple_span_uses_ceiling() -> None: # Span = 16 min (12:00 → 12:16): ticks 12:00 AND 12:15 are < 12:16. # floor(16/15) = 1 would under-count; the correct answer is 2. since = datetime(2026, 5, 12, 12, 0, tzinfo=UTC) until = datetime(2026, 5, 12, 12, 16, tzinfo=UTC) assert _expected_ticks(since, until) == 2 def test_expected_ticks_exact_quarter_boundary_is_excluded() -> None: # Until landing exactly on a */15 tick: that tick is NOT counted # because the window is half-open on the right. since = datetime(2026, 5, 12, 12, 0, tzinfo=UTC) until = datetime(2026, 5, 12, 12, 45, tzinfo=UTC) # Ticks at 12:00, 12:15, 12:30 → 3 (12:45 excluded) assert _expected_ticks(since, until) == 3