Odysseus v1.0

2026-05-31 23:58:26 +09:00
commit e5c99a5eee
421 changed files with 271349 additions and 0 deletions
--- a/tests/test_search_ranking.py
+++ b/tests/test_search_ranking.py
@@ -0,0 +1,39 @@
+from src.search.ranking import rank_search_results
+
+
+def test_news_queries_prefer_news_sources_over_sports_and_social_results():
+    results = [
+        {
+            "title": "Chicago Stars fire GM Richard Feuz",
+            "url": "https://www.reuters.com/sports/soccer/chicago-stars-fire-gm-richard-feuz--flm-2026-05-27/",
+            "snippet": "The Chicago Stars fired their general manager.",
+        },
+        {
+            "title": "United States Eliminates Canada In Quarterfinals",
+            "url": "https://sports.yahoo.com/articles/united-states-vs-canada-live-updates-170747222.html",
+            "snippet": "United States eliminated Canada in hockey.",
+        },
+        {
+            "title": "Canada - AP News",
+            "url": "https://apnews.com/hub/canada",
+            "snippet": "Stay up to date on the latest Canada news coverage from AP News.",
+        },
+        {
+            "title": "CBC News - Canada",
+            "url": "https://www.cbc.ca/news/canada",
+            "snippet": "Your source for Canadian news in English.",
+        },
+        {
+            "title": "CTV News - Canada",
+            "url": "https://www.ctvnews.ca/canada",
+            "snippet": "Latest news, travel, politics, money, jobs and more.",
+        },
+    ]
+
+    ranked = rank_search_results("Canada news today", results)
+    top_urls = [item["url"] for item in ranked[:3]]
+
+    assert "https://apnews.com/hub/canada" in top_urls
+    assert "https://www.cbc.ca/news/canada" in top_urls
+    assert "https://www.ctvnews.ca/canada" in top_urls
+    assert ranked[-1]["url"].startswith("https://sports.yahoo.com/")