6
6
import pytest
7
7
import pandas as pd
8
8
from dotenv import load_dotenv
9
- from scrapegraphai .graphs import SmartScraperMultiParseConcatFirstGraph
9
+ from scrapegraphai .graphs import SmartScraperMultiLiteGraph
10
10
from scrapegraphai .utils import prettify_exec_info
11
11
12
12
load_dotenv ()
@@ -27,7 +27,7 @@ def graph_config():
27
27
28
28
def test_scraping_pipeline (graph_config ):
29
29
"""Start of the scraping pipeline"""
30
- smart_scraper_multi_parse_concat_first_graph = SmartScraperMultiParseConcatFirstGraph (
30
+ smart_scraper_multi_lite_graph = SmartScraperMultiLiteGraph (
31
31
prompt = "Who is Marco Perini?" ,
32
32
source = [
33
33
"https://perinim.github.io/" ,
@@ -36,14 +36,14 @@ def test_scraping_pipeline(graph_config):
36
36
config = graph_config ,
37
37
)
38
38
39
- result = smart_scraper_multi_parse_concat_first_graph .run ()
39
+ result = smart_scraper_multi_lite_graph .run ()
40
40
41
41
assert result is not None
42
42
assert isinstance (result , dict )
43
43
44
44
def test_get_execution_info (graph_config ):
45
45
"""Get the execution info"""
46
- smart_scraper_multi_parse_concat_first_graph = SmartScraperMultiParseConcatFirstGraph (
46
+ smart_scraper_multi_lite_graph = SmartScraperMultiLiteGraph (
47
47
prompt = "Who is Marco Perini?" ,
48
48
source = [
49
49
"https://perinim.github.io/" ,
@@ -52,8 +52,8 @@ def test_get_execution_info(graph_config):
52
52
config = graph_config ,
53
53
)
54
54
55
- smart_scraper_multi_parse_concat_first_graph .run ()
55
+ smart_scraper_multi_lite_graph .run ()
56
56
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 ()
58
58
59
59
assert graph_exec_info is not None
0 commit comments