45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
import json, pytest
|
|
from tests.services.recipe.test_element_classifier import store_with_profiles
|
|
|
|
|
|
@pytest.fixture
|
|
def store_with_subs(store_with_profiles):
|
|
store_with_profiles.conn.execute("""
|
|
INSERT INTO substitution_pairs
|
|
(original_name, substitute_name, constraint_label,
|
|
fat_delta, moisture_delta, glutamate_delta, occurrence_count)
|
|
VALUES (?,?,?,?,?,?,?)
|
|
""", ("butter", "coconut oil", "vegan", -1.0, 0.0, 0.0, 15))
|
|
store_with_profiles.conn.execute("""
|
|
INSERT INTO substitution_pairs
|
|
(original_name, substitute_name, constraint_label,
|
|
fat_delta, moisture_delta, glutamate_delta, occurrence_count)
|
|
VALUES (?,?,?,?,?,?,?)
|
|
""", ("ground beef", "lentils", "vegan", -15.0, 10.0, -2.0, 45))
|
|
store_with_profiles.conn.commit()
|
|
return store_with_profiles
|
|
|
|
|
|
def test_find_substitutes_for_constraint(store_with_subs):
|
|
from app.services.recipe.substitution_engine import SubstitutionEngine
|
|
engine = SubstitutionEngine(store_with_subs)
|
|
results = engine.find_substitutes("butter", constraint="vegan")
|
|
assert len(results) > 0
|
|
assert results[0].substitute_name == "coconut oil"
|
|
|
|
|
|
def test_compensation_hints_for_large_delta(store_with_subs):
|
|
from app.services.recipe.substitution_engine import SubstitutionEngine
|
|
engine = SubstitutionEngine(store_with_subs)
|
|
results = engine.find_substitutes("ground beef", constraint="vegan")
|
|
assert len(results) > 0
|
|
swap = results[0]
|
|
# Fat delta is -15g — should suggest a Richness compensation
|
|
assert any(h["element"] == "Richness" for h in swap.compensation_hints)
|
|
|
|
|
|
def test_no_substitutes_returns_empty(store_with_subs):
|
|
from app.services.recipe.substitution_engine import SubstitutionEngine
|
|
engine = SubstitutionEngine(store_with_subs)
|
|
results = engine.find_substitutes("unobtainium", constraint="vegan")
|
|
assert results == []
|