Skip to content

Commit b9028de

Browse files
committed
Squeeze results to enable string indexing
1 parent b238b08 commit b9028de

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphdatascience/tests/integration/test_remote_graph_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,10 @@ def test_remote_projection_and_writeback_custom_database_name(gds_with_cloud_set
6060
write_result = gds_with_cloud_setup.wcc.write(G, writeProperty="wcc")
6161

6262
assert write_result["nodePropertiesWritten"] == 2
63-
db_state = gds_with_cloud_setup.run_cypher("MATCH (n WHERE n.wcc IS NOT NULL) RETURN count(*) AS c")
63+
db_state = gds_with_cloud_setup.run_cypher("MATCH (n WHERE n.wcc IS NOT NULL) RETURN count(*) AS c").squeeze()
6464
assert db_state["c"] == 2
6565
gds_with_cloud_setup.set_database("neo4j")
66-
other_db_state = gds_with_cloud_setup.run_cypher("MATCH (n WHERE n.wcc IS NOT NULL) RETURN count(*) AS c")
66+
other_db_state = gds_with_cloud_setup.run_cypher("MATCH (n WHERE n.wcc IS NOT NULL) RETURN count(*) AS c").squeeze()
6767
assert other_db_state["c"] == 0
6868

6969

0 commit comments

Comments
 (0)