diet-guard/diet_guard/tests/test_portions.py
Krzysztof kuhy Rudnicki 400f89b469 feat(diet_guard): add meal-logging screen-lock gate with trigger fix
Add the diet_guard package: a screen-locking meal-logging gate that fires
on 4-hour slots (08/12/16/20) and records calories/macros, persisting an
autocompleting food bank.

- Trigger fix: the systemd timer fires at session start (Persistent=true)
  before lightdm has written ~/.Xauthority, so the gate crashed with a
  TclError instead of locking the screen. Add wait_for_display() /
  _display_is_ready() in _gatelock.py and wire it into _cli._cmd_gate so the
  gate retries on the next tick instead of crashing; add
  Environment=XAUTHORITY=%h/.Xauthority to the service as belt-and-suspenders.
- Food-bank hardening: a transiently corrupt food_bank.json was warned about
  on every keystroke and then silently overwritten (data loss). _read_bank
  now quarantines it via _quarantine_corrupt_bank() (warn-once + timestamped
  backup) before starting fresh.
- Multi-item meals: new _meal.py (MealItem, meal_total, MEAL_SOURCE),
  remember_meal() + _upsert() in _foodbank.py, and a "+ Add item" control in
  the gate that logs both the individual items and the composite meal.
- Bundle resolve_nutrition's manual macros into a ManualMacros dataclass to
  stay within the argument-count limit.

diet_guard at 100% branch coverage; full pre-commit suite passes.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-10 22:32:39 +02:00

66 lines
2.1 KiB
Python

"""Tests for _portions.py — built-in staple weights and macros.
Covers the fuzzy staple match, the empty-input guard, and the per-100 g
Nutrition / suggestion builders.
"""
from __future__ import annotations
from python_pkg.diet_guard._portions import (
estimate_unit_grams,
staple_nutrition,
suggest_staples,
)
class TestEstimateUnitGrams:
"""One piece's typical weight."""
def test_known_staple(self) -> None:
"""A known staple returns its curated unit weight."""
assert estimate_unit_grams("apple") == 182.0
def test_fuzzy_plural(self) -> None:
"""A close variant (plural) still matches the staple."""
assert estimate_unit_grams("apples") == 182.0
def test_unknown_returns_none(self) -> None:
"""An unrecognised food has no known unit weight."""
assert estimate_unit_grams("quinoa risotto") is None
def test_empty_returns_none(self) -> None:
"""A blank description short-circuits to None."""
assert estimate_unit_grams(" ") is None
class TestStapleNutrition:
"""Per-100 g Nutrition for a staple."""
def test_known_staple_per_100g(self) -> None:
"""An egg resolves to its per-100 g macros at a 100 g basis."""
nutrition = staple_nutrition("egg")
assert nutrition is not None
assert nutrition.grams == 100.0
assert nutrition.source == "staple: egg"
def test_unknown_returns_none(self) -> None:
"""A non-staple resolves to None."""
assert staple_nutrition("beef wellington") is None
class TestSuggestStaples:
"""Live autocomplete over the staple table."""
def test_match(self) -> None:
"""A matching query surfaces the staple by name."""
names = [name for name, _ in suggest_staples("banana")]
assert "banana" in names
def test_empty_query(self) -> None:
"""A blank query suggests nothing."""
assert suggest_staples("") == []
def test_no_match(self) -> None:
"""A query matching no staple returns an empty list."""
assert suggest_staples("xyzzy") == []