From 113a2398fcb021aa44d756ebc13fa8efcb969c99 Mon Sep 17 00:00:00 2001 From: Logan Markewich Date: Tue, 21 Jan 2025 19:08:33 -0600 Subject: [PATCH] nit --- .../tests/test_pg_stores_memgraph.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/llama-index-integrations/graph_stores/llama-index-graph-stores-memgraph/tests/test_pg_stores_memgraph.py b/llama-index-integrations/graph_stores/llama-index-graph-stores-memgraph/tests/test_pg_stores_memgraph.py index 8045c040a5ce3..4b808b4cf7266 100644 --- a/llama-index-integrations/graph_stores/llama-index-graph-stores-memgraph/tests/test_pg_stores_memgraph.py +++ b/llama-index-integrations/graph_stores/llama-index-graph-stores-memgraph/tests/test_pg_stores_memgraph.py @@ -81,14 +81,13 @@ def test_memgraph_pg_store(pg_store: MemgraphPropertyGraphStore) -> None: label="PERSON", name="Logan", properties={"age": 28, "location": "Canada"} ) pg_store.upsert_nodes([new_node]) - + # Assert the node has been updated with the new property kg_nodes = pg_store.get(properties={"age": 28}) - + # Test deleting nodes from Memgraph. pg_store.delete(ids=[source_node.node_id]) pg_store.delete(ids=[entity1.id, entity2.id]) # Assert the nodes have been deleted pg_store.get(ids=[entity1.id, entity2.id]) - \ No newline at end of file