| from app.agents.research_note_graph import classify_topic_sensitivity, research_topic | |
| def test_classifies_sensitive_health_topic(): | |
| profile = classify_topic_sensitivity("Auditory hallucinations in medical science") | |
| assert profile["topic_domain"] == "health_science" | |
| assert profile["sensitivity_level"] == "high" | |
| def test_research_topic_uses_wikipedia(monkeypatch): | |
| monkeypatch.setattr( | |
| "app.agents.research_note_graph._wikipedia_summary", | |
| lambda query: ( | |
| [ | |
| { | |
| "text": "Auditory hallucinations are perceptions of sound without an external acoustic stimulus.", | |
| "score": 1.0, | |
| "topic": query, | |
| "source": "Wikipedia: Auditory hallucination", | |
| "url": "https://en.wikipedia.org/wiki/Auditory_hallucination", | |
| "kind": "research", | |
| } | |
| ], | |
| None, | |
| ), | |
| ) | |
| result = research_topic("auditory hallucinations") | |
| assert result["sources"][0]["source"] == "Wikipedia: Auditory hallucination" | |
| assert "Auditory hallucinations" in result["sources"][0]["text"] | |