44
55from graphdatascience import ServerVersion
66from graphdatascience .call_parameters import CallParameters
7- from graphdatascience .query_runner .aura_db_arrow_query_runner import (
8- AuraDbArrowQueryRunner ,
9- )
7+ from graphdatascience .query_runner .aura_db_query_runner import AuraDbQueryRunner
108from graphdatascience .tests .unit .conftest import CollectingQueryRunner
119
1210
@@ -24,7 +22,7 @@ def test_extracts_parameters_projection() -> None:
2422 db_query_runner = CollectingQueryRunner (version )
2523 gds_query_runner = CollectingQueryRunner (version )
2624 gds_query_runner .set__mock_result (DataFrame ([{"databaseLocation" : "remote" }]))
27- qr = AuraDbArrowQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
25+ qr = AuraDbQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
2826
2927 qr .call_procedure (
3028 endpoint = "gds.arrow.project" ,
@@ -66,7 +64,7 @@ def test_extracts_parameters_algo_write() -> None:
6664 db_query_runner = CollectingQueryRunner (version )
6765 gds_query_runner = CollectingQueryRunner (version )
6866 gds_query_runner .set__mock_result (DataFrame ([{"databaseLocation" : "remote" }]))
69- qr = AuraDbArrowQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
67+ qr = AuraDbQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
7068
7169 qr .call_procedure (endpoint = "gds.degree.write" , params = CallParameters (graph_name = "g" , config = {}))
7270
@@ -92,7 +90,7 @@ def test_arrow_and_write_configuration() -> None:
9290 db_query_runner = CollectingQueryRunner (version )
9391 gds_query_runner = CollectingQueryRunner (version )
9492 gds_query_runner .set__mock_result (DataFrame ([{"databaseLocation" : "remote" }]))
95- qr = AuraDbArrowQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
93+ qr = AuraDbQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
9694
9795 qr .call_procedure (
9896 endpoint = "gds.degree.write" ,
@@ -130,7 +128,7 @@ def test_arrow_and_write_configuration_graph_write() -> None:
130128 db_query_runner = CollectingQueryRunner (version )
131129 gds_query_runner = CollectingQueryRunner (version )
132130 gds_query_runner .set__mock_result (DataFrame ([{"databaseLocation" : "remote" }]))
133- qr = AuraDbArrowQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
131+ qr = AuraDbQueryRunner (gds_query_runner , db_query_runner , FakeArrowClient (), False ) # type: ignore
134132
135133 qr .call_procedure (
136134 endpoint = "gds.graph.nodeProperties.write" ,
0 commit comments