mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-07-09 21:19:20 +08:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import pytest
|
|
|
|
from scrapegraphai.utils.research_web import ( # Replace with actual path to your file
|
|
search_on_web,
|
|
)
|
|
|
|
|
|
def test_google_search():
|
|
"""Tests search_on_web with Google search engine."""
|
|
results = search_on_web("test query", search_engine="Google", max_results=2)
|
|
assert len(results) == 2
|
|
# You can further assert if the results actually contain 'test query' in the title/snippet using additional libraries
|
|
|
|
|
|
def test_bing_search():
|
|
"""Tests search_on_web with Bing search engine."""
|
|
results = search_on_web("test query", search_engine="Bing", max_results=1)
|
|
assert results is not None
|
|
# You can further assert if the results contain '.com' or '.org' in the domain
|
|
|
|
|
|
def test_invalid_search_engine():
|
|
"""Tests search_on_web with invalid search engine."""
|
|
with pytest.raises(ValueError):
|
|
search_on_web("test query", search_engine="Yahoo", max_results=5)
|
|
|
|
|
|
def test_max_results():
|
|
"""Tests search_on_web with different max_results values."""
|
|
results_5 = search_on_web("test query", max_results=5)
|
|
results_10 = search_on_web("test query", max_results=10)
|
|
assert len(results_5) <= len(results_10)
|