Skip to content

Commit 1dbdaa3

Browse files
committed
Add arrow_configuration to remote projection unit tests
1 parent bc63ed5 commit 1dbdaa3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

graphdatascience/tests/unit/test_graph_ops.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def test_project_remote(runner: CollectingQueryRunner, aura_gds: AuraGraphDataSc
9797
assert (
9898
runner.last_query()
9999
== "CALL gds.arrow.project("
100-
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types)"
100+
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
101101
)
102102
# injection of token and host into the params is done by the actual query runner
103103
assert runner.last_params() == {
@@ -106,6 +106,7 @@ def test_project_remote(runner: CollectingQueryRunner, aura_gds: AuraGraphDataSc
106106
"inverse_indexed_relationship_types": [],
107107
"query": "RETURN gds.graph.project.remote(0, 1, null)",
108108
"undirected_relationship_types": [],
109+
"arrow_configuration": {},
109110
}
110111

111112

@@ -722,7 +723,7 @@ def test_remote_projection_all_configuration(runner: CollectingQueryRunner, aura
722723
assert (
723724
runner.last_query()
724725
== "CALL gds.arrow.project("
725-
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types)"
726+
+ "$graph_name, $query, $concurrency, $undirected_relationship_types, $inverse_indexed_relationship_types, $arrow_configuration)"
726727
)
727728

728729
assert runner.last_params() == {
@@ -738,4 +739,5 @@ def test_remote_projection_all_configuration(runner: CollectingQueryRunner, aura
738739
""",
739740
"undirected_relationship_types": ["R"],
740741
"inverse_indexed_relationship_types": ["R"],
742+
"arrow_configuration": {},
741743
}

0 commit comments

Comments
 (0)