CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-09 21:19:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: Update coverage improvement test for tests/test_search_graph.py

Browse Source
This commit is contained in:
codebeaver-ai[bot] 2025-01-29 09:34:27 +00:00 committed by GitHub
parent f5ebe8ac10
commit ba58568b8a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/test_search_graph.py
View File

@ -1,7 +1,7 @@
import pytest
from scrapegraphai.graphs.search_graph import SearchGraph
from unittest.mock import MagicMock, patch
from unittest.mock import MagicMock, call, patch
class TestSearchGraph:
"""Test class for SearchGraph"""
@ -57,4 +57,26 @@ class TestSearchGraph:
result = search_graph.run()
# Assert
assert result == "No answer found."
assert result == "No answer found."
@patch('scrapegraphai.graphs.search_graph.SearchInternetNode')
@patch('scrapegraphai.graphs.search_graph.GraphIteratorNode')
@patch('scrapegraphai.graphs.search_graph.MergeAnswersNode')
@patch('scrapegraphai.graphs.search_graph.BaseGraph')
@patch('scrapegraphai.graphs.abstract_graph.AbstractGraph._create_llm')
def test_max_results_config(self, mock_create_llm, mock_base_graph, mock_merge_answers, mock_graph_iterator, mock_search_internet):
"""
Test that the max_results parameter from the config is correctly passed to the SearchInternetNode.
"""
# Arrange
prompt = "Test prompt"
max_results = 5
config = {"llm": {"model": "test-model"}, "max_results": max_results}
# Act
search_graph = SearchGraph(prompt, config)
# Assert
mock_search_internet.assert_called_once()
call_args = mock_search_internet.call_args
assert call_args.kwargs['node_config']['max_results'] == max_results