Spaces:
Running on Zero
Running on Zero
File size: 2,265 Bytes
f00d669 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | """Brick 5 tests: persistent taste profile blended with trip mood (P1-1)."""
from __future__ import annotations
import pytest
from discoverroute import config
pytest.importorskip("sentence_transformers")
from discoverroute.interpret import profile as prof
def test_empty_profile_has_no_affinity():
assert prof.profile_affinity(prof.empty_profile()) is None
assert prof.profile_affinity({}) is None
def test_saved_places_boost_their_categories():
p = {"standing_text": "", "saved_categories": ["park_garden", "park_garden"]}
aff = prof.profile_affinity(p)
assert aff["park_garden"] > aff["bar_pub"]
def test_standing_text_shapes_affinity():
p = {"standing_text": "I love quiet bookshops and libraries", "saved_categories": []}
aff = prof.profile_affinity(p)
assert aff["bookshop"] > aff["bar_pub"]
assert aff["library"] > aff["restaurant"]
def test_effective_blend_modes():
park_profile = {"standing_text": "", "saved_categories": ["park_garden"]}
# profile only
w_prof = prof.effective_weights(park_profile, trip_vibe="")
assert w_prof.category_affinity["park_garden"] > w_prof.category_affinity["bar_pub"]
# trip mood only (no profile)
w_trip = prof.effective_weights({}, trip_vibe="lively bars and nightlife")
assert w_trip.category_affinity["bar_pub"] > w_trip.category_affinity["park_garden"]
# neither -> uniform
w_none = prof.effective_weights({}, trip_vibe="")
assert set(w_none.category_affinity.values()) == {1.0}
data_ready = pytest.mark.skipif(
not (config.GRAPH_WALK_PATH.exists() and config.POIS_PATH.exists()),
reason="Graph or POI table not built",
)
@data_ready
def test_profile_shifts_route():
"""Editing the profile measurably shifts the route (the P1-1 DoD)."""
from discoverroute.pipeline import plan_route
a, b = "Place de la République, Paris", "Jardin du Luxembourg, Paris"
none = plan_route(a, b, budget=0.7)
booky = plan_route(a, b, budget=0.7,
profile={"standing_text": "quiet libraries and bookshops",
"saved_categories": ["library", "bookshop"]})
assert booky.discovery is not None
assert {p.osm_id for p in none.pois} != {p.osm_id for p in booky.pois}
|