mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-09-19 06:32:20 +02:00
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
from services.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/")
|
|
|
|
|
|
def test_software_release_queries_prefer_official_github_releases():
|
|
results = [
|
|
{
|
|
"title": "ollama/ollama on GitHub | Release Alert",
|
|
"url": "https://releasealert.dev/github/ollama/ollama",
|
|
"snippet": "Latest releases for ollama/ollama on GitHub.",
|
|
},
|
|
{
|
|
"title": "Ollama Release Notes - August 2026 Latest Updates",
|
|
"url": "https://releasebot.io/updates/ollama",
|
|
"snippet": "Complete list of Ollama latest updates.",
|
|
},
|
|
{
|
|
"title": "Releases · ollama/ollama - GitHub",
|
|
"url": "https://github.com/ollama/ollama/releases",
|
|
"snippet": "Latest release v0.33.0.",
|
|
},
|
|
]
|
|
|
|
ranked = rank_search_results("latest ollama release version github", results)
|
|
|
|
assert ranked[0]["url"] == "https://github.com/ollama/ollama/releases"
|
|
|
|
|
|
def test_product_spec_queries_prefer_official_shop_or_specs_pages():
|
|
results = [
|
|
{
|
|
"title": "Surprise launch coverage for new creator laptop",
|
|
"url": "https://example-news.com/tech/surprise-launch-creator-laptop",
|
|
"snippet": "The laptop was announced yesterday and may ship later this year.",
|
|
},
|
|
{
|
|
"title": "Creator Laptop - Buy",
|
|
"url": "https://www.creatorlaptop.com/shop/buy-laptop",
|
|
"snippet": "Configure with 64GB RAM. See pricing, availability, and shipping options.",
|
|
},
|
|
{
|
|
"title": "Creator Laptop rumors and expected specs",
|
|
"url": "https://rumor.example.com/creator-laptop-expected-specs",
|
|
"snippet": "Rumored future models could include more memory.",
|
|
},
|
|
]
|
|
|
|
ranked = rank_search_results("current creator laptop memory price availability", results)
|
|
|
|
assert ranked[0]["url"] == "https://www.creatorlaptop.com/shop/buy-laptop"
|