Skip to content

Commit 28dda2b

Browse files
committed
rename graph name
1 parent 3e8f047 commit 28dda2b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/graphs/smart_scraper_multi_parse_merge_first_graph_openai_test.py renamed to tests/graphs/smart_scraper_multi_lite_graph_openai_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import pytest
77
import pandas as pd
88
from dotenv import load_dotenv
9-
from scrapegraphai.graphs import SmartScraperMultiParseConcatFirstGraph
9+
from scrapegraphai.graphs import SmartScraperMultiLiteGraph
1010
from scrapegraphai.utils import prettify_exec_info
1111

1212
load_dotenv()
@@ -27,7 +27,7 @@ def graph_config():
2727

2828
def test_scraping_pipeline(graph_config):
2929
"""Start of the scraping pipeline"""
30-
smart_scraper_multi_parse_concat_first_graph = SmartScraperMultiParseConcatFirstGraph(
30+
smart_scraper_multi_lite_graph = SmartScraperMultiLiteGraph(
3131
prompt="Who is Marco Perini?",
3232
source= [
3333
"https://perinim.github.io/",
@@ -36,14 +36,14 @@ def test_scraping_pipeline(graph_config):
3636
config=graph_config,
3737
)
3838

39-
result = smart_scraper_multi_parse_concat_first_graph.run()
39+
result = smart_scraper_multi_lite_graph.run()
4040

4141
assert result is not None
4242
assert isinstance(result, dict)
4343

4444
def test_get_execution_info(graph_config):
4545
"""Get the execution info"""
46-
smart_scraper_multi_parse_concat_first_graph = SmartScraperMultiParseConcatFirstGraph(
46+
smart_scraper_multi_lite_graph = SmartScraperMultiLiteGraph(
4747
prompt="Who is Marco Perini?",
4848
source= [
4949
"https://perinim.github.io/",
@@ -52,8 +52,8 @@ def test_get_execution_info(graph_config):
5252
config=graph_config,
5353
)
5454

55-
smart_scraper_multi_parse_concat_first_graph.run()
55+
smart_scraper_multi_lite_graph.run()
5656

57-
graph_exec_info = smart_scraper_multi_parse_concat_first_graph.get_execution_info()
57+
graph_exec_info = smart_scraper_multi_lite_graph.get_execution_info()
5858

5959
assert graph_exec_info is not None

0 commit comments

Comments
 (0)