66import pytest
77import pandas as pd
88from dotenv import load_dotenv
9- from scrapegraphai .graphs import SmartScraperMultiParseConcatFirstGraph
9+ from scrapegraphai .graphs import SmartScraperMultiLiteGraph
1010from scrapegraphai .utils import prettify_exec_info
1111
1212load_dotenv ()
@@ -27,7 +27,7 @@ def graph_config():
2727
2828def 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
4444def 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