Skip to content

Commit 4de8cf8

Browse files
authored
Tests: fix type hints (#1030)
1 parent b9dfeb5 commit 4de8cf8

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/unit/async_/test_driver.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1012,8 +1012,9 @@ async def test_execute_query_result_transformer(
10121012
driver = AsyncGraphDatabase.driver("bolt://localhost")
10131013
res: t.Any
10141014
async with driver as driver:
1015+
expected_transformer: t.Any = result_transformer
10151016
if result_transformer is Ellipsis:
1016-
result_transformer = AsyncResult.to_eager_result
1017+
expected_transformer = AsyncResult.to_eager_result
10171018
res_default: neo4j.EagerResult = await driver.execute_query("")
10181019
res = res_default
10191020
else:
@@ -1036,7 +1037,7 @@ async def test_execute_query_result_transformer(
10361037
session_executor_mock = session_mock._run_transaction
10371038
session_executor_mock.assert_awaited_once_with(
10381039
WRITE_ACCESS, TelemetryAPI.DRIVER, _work,
1039-
(mocker.ANY, mocker.ANY, result_transformer), {}
1040+
(mocker.ANY, mocker.ANY, expected_transformer), {}
10401041
)
10411042
assert res is session_executor_mock.return_value
10421043

tests/unit/sync/test_driver.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1011,8 +1011,9 @@ def test_execute_query_result_transformer(
10111011
driver = GraphDatabase.driver("bolt://localhost")
10121012
res: t.Any
10131013
with driver as driver:
1014+
expected_transformer: t.Any = result_transformer
10141015
if result_transformer is Ellipsis:
1015-
result_transformer = Result.to_eager_result
1016+
expected_transformer = Result.to_eager_result
10161017
res_default: neo4j.EagerResult = driver.execute_query("")
10171018
res = res_default
10181019
else:
@@ -1035,7 +1036,7 @@ def test_execute_query_result_transformer(
10351036
session_executor_mock = session_mock._run_transaction
10361037
session_executor_mock.assert_called_once_with(
10371038
WRITE_ACCESS, TelemetryAPI.DRIVER, _work,
1038-
(mocker.ANY, mocker.ANY, result_transformer), {}
1039+
(mocker.ANY, mocker.ANY, expected_transformer), {}
10391040
)
10401041
assert res is session_executor_mock.return_value
10411042

0 commit comments

Comments
 (0)