From 3a537eec6fef1743924a9aa5cef0ba2f8d44bf11 Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Sun, 23 Jun 2024 17:54:09 +0200
Subject: [PATCH] fix: add test

---
 scrapegraphai/utils/parse_state_keys.py | 15 ---------------
 tests/utils/parse_state_keys_test.py    | 21 +++++++++++++++++++++
 2 files changed, 21 insertions(+), 15 deletions(-)
 create mode 100644 tests/utils/parse_state_keys_test.py

diff --git a/scrapegraphai/utils/parse_state_keys.py b/scrapegraphai/utils/parse_state_keys.py
index 6afc2ecb..85712ef6 100644
--- a/scrapegraphai/utils/parse_state_keys.py
+++ b/scrapegraphai/utils/parse_state_keys.py
@@ -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)
diff --git a/tests/utils/parse_state_keys_test.py b/tests/utils/parse_state_keys_test.py
new file mode 100644
index 00000000..d91355f1
--- /dev/null
+++ b/tests/utils/parse_state_keys_test.py
@@ -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)