Spaces:
Running
Running
| from datetime import date, timedelta | |
| from focus_buddy.memory import RECENT_WINDOW, DayMemory | |
| from focus_buddy.observations import Habit, Observation | |
| BITING = Observation(biting_nails=True) | |
| IDLE = Observation() | |
| class TestEpisodeCounting: | |
| def test_counts_episodes_not_frames(self): | |
| # At 1fps a 30-second habit is one episode, not thirty. | |
| memory = DayMemory() | |
| for _ in range(30): | |
| memory.record(BITING) | |
| assert memory.counters[Habit.BITING_NAILS] == 1 | |
| def test_counts_a_new_episode_after_a_gap(self): | |
| memory = DayMemory() | |
| memory.record(BITING) | |
| memory.record(IDLE) | |
| memory.record(BITING) | |
| assert memory.counters[Habit.BITING_NAILS] == 2 | |
| def test_recent_window_is_bounded(self): | |
| memory = DayMemory() | |
| for _ in range(RECENT_WINDOW * 3): | |
| memory.record(IDLE) | |
| assert len(memory.recent) == RECENT_WINDOW | |
| class TestDayRollover: | |
| def test_rollover_resets_counters(self): | |
| memory = DayMemory() | |
| memory.record(BITING) | |
| memory.day = date.today() - timedelta(days=1) | |
| assert memory.roll_day_if_needed() is True | |
| assert memory.counters[Habit.BITING_NAILS] == 0 | |
| assert memory.day == date.today() | |
| def test_no_rollover_within_the_same_day(self): | |
| assert DayMemory().roll_day_if_needed() is False | |
| def test_episode_state_resets_across_the_rollover(self): | |
| # Without resetting the active-habit map, a habit still in progress at | |
| # midnight would never be counted again on the new day. | |
| memory = DayMemory() | |
| memory.record(BITING) | |
| memory.day = date.today() - timedelta(days=1) | |
| memory.roll_day_if_needed() | |
| memory.record(BITING) | |
| assert memory.counters[Habit.BITING_NAILS] == 1 | |
| class TestSummary: | |
| def test_mentions_every_habit_and_the_elapsed_time(self): | |
| clock = iter([0.0] + [3600.0] * 20) | |
| memory = DayMemory(clock=lambda: next(clock)) | |
| memory.counters[Habit.BITING_NAILS] = 3 | |
| memory.counters[Habit.LOOKING_AT_PHONE] = 1 | |
| summary = memory.summary() | |
| assert "nail-biting 3 times" in summary | |
| assert "phone 1 time" in summary # singular | |
| assert "You've been working 1h" in summary | |