Skip to content

Commit 76f2839

Browse files
DarthMaxFlorentinD
authored andcommitted
Fix unit tests for AuraGraphDataScience
1 parent 4c1923b commit 76f2839

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

graphdatascience/tests/unit/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,8 @@ def aura_gds(runner: CollectingQueryRunner, mocker: MockerFixture) -> Generator[
221221
mocker.patch("graphdatascience.query_runner.session_query_runner.SessionQueryRunner.create", return_value=runner)
222222
mocker.patch("graphdatascience.query_runner.arrow_query_runner.ArrowQueryRunner.create", return_value=runner)
223223
mocker.patch("graphdatascience.query_runner.gds_arrow_client.GdsArrowClient.create", return_value=None)
224+
mocker.patch("graphdatascience.arrow_client.authenticated_flight_client.AuthenticatedArrowClient.create", return_value=None)
225+
mocker.patch("graphdatascience.session.session_v2_endpoints.SessionV2Endpoints.__init__", return_value=None)
224226

225227
aura_gds = AuraGraphDataScience.create(
226228
session_bolt_connection_info=DbmsConnectionInfo("address", "some", "auth"),

0 commit comments

Comments
 (0)