Skip to content

Commit c11128b

Browse files
committed
Rename class and test file after rebase
1 parent 6b90b08 commit c11128b

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

graphdatascience/tests/unit/test_aura_db_arrow_query_runner.py renamed to graphdatascience/tests/unit/test_aura_db_query_runner.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,7 @@
44

55
from graphdatascience import ServerVersion
66
from 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
108
from 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

Comments
 (0)