File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments