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

fix: add test

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-06-23 17:54:09 +02:00
parent 79a2f51c34
commit 3a537eec6f
2 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scrapegraphai/utils/parse_state_keys.py
View File

@ -101,18 +101,3 @@ def parse_expression(expression, state: dict) -> list:
final_result.append(key)
return final_result
EXPRESSION = "user_input & (relevant_chunks | parsed_document | document)"
state = {
"user_input": None,
"document": None,
"parsed_document": None,
"relevant_chunks": None,
}
try:
result = parse_expression(EXPRESSION, state)
print("Matched keys:", result)
except ValueError as e:
print("Error:", e)

21
tests/utils/parse_state_keys_test.py Normal file
View File

@ -0,0 +1,21 @@
"""
Parse_state_key test module
"""
import pytest
from scrapegraphai.utils.parse_state_keys import parse_expression
def test_parse_expression():
"""Test parse_expression function."""
EXPRESSION = "user_input & (relevant_chunks | parsed_document | document)"
state = {
"user_input": None,
"document": None,
"parsed_document": None,
"relevant_chunks": None,
}
try:
result = parse_expression(EXPRESSION, state)
assert result != []
except ValueError as e:
assert "Error" in str(e)