mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-07-06 21:11:37 +08:00
24 lines
561 B
Python
24 lines
561 B
Python
"""
|
|
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)
|