@@ -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