Add UT to make sure we're not pruning the lexical graph

Author: stellasia

tests/unit/experimental/components/test_graph_pruning.py

@@ -39,7 +39,9 @@
 
 @pytest.fixture(scope="module")
 def lexical_graph_config() -> LexicalGraphConfig:
-    return LexicalGraphConfig()
+    return LexicalGraphConfig(
+        chunk_node_label="Paragraph",
+    )
 
 
 @pytest.mark.parametrize(
@@ -165,6 +167,20 @@ def test_graph_pruning_validate_node(
         assert result is None
 
 
+def test_graph_pruning_enforce_nodes_lexical_graph(lexical_graph_config: LexicalGraphConfig) -> None:
+    pruner = GraphPruning()
+    result = pruner._enforce_nodes(
+        extracted_nodes=[
+            Neo4jNode(id="1", label="Paragraph"),
+        ],
+        schema=GraphSchema(additional_node_types=False),
+        lexical_graph_config=lexical_graph_config,
+        pruning_stats=PruningStats(),
+    )
+    assert len(result) == 1
+    assert result[0].label == "Paragraph"
+
+
 @pytest.fixture
 def neo4j_relationship() -> Neo4jRelationship:
     return Neo4jRelationship(