From 74dcdb417be43b293583030850eb8dabee6ab5a4 Mon Sep 17 00:00:00 2001 From: Devin Stein Date: Tue, 12 Nov 2024 21:05:40 -0700 Subject: [PATCH] fix(tests): use assert_called_once_with for 3.12 compat --- sparkmagic/sparkmagic/tests/test_sparkkernelbase.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py b/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py index cc557c5f..6e149a5a 100644 --- a/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py +++ b/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py @@ -45,11 +45,11 @@ def test_execute_valid_code(): ret = kernel.do_execute(code, False) kernel.user_code_parser.get_code_to_run.assert_called_once_with(code) - assert execute_cell_mock.called_once_with(ret, True) + assert execute_cell_mock.assert_called_once_with(ret, True) assert execute_cell_mock.return_value is ret assert kernel._fatal_error is None - assert execute_cell_mock.called_once_with(code, True) + assert execute_cell_mock.assert_called_once_with(code, True) assert ipython_display.send_error.call_count == 0 @@ -62,7 +62,7 @@ def test_execute_throws_if_fatal_error_happened(): assert execute_cell_mock.return_value is ret assert kernel._fatal_error == fatal_error - assert execute_cell_mock.called_once_with("None", True) + assert execute_cell_mock.assert_called_once_with("None", True) assert ipython_display.send_error.call_count == 1 @@ -74,7 +74,7 @@ def test_execute_alerts_user_if_an_unexpected_error_happens(): ret = kernel.do_execute(code, False) assert execute_cell_mock.return_value is ret - assert execute_cell_mock.called_once_with("None", True) + assert execute_cell_mock.assert_called_once_with("None", True) assert ipython_display.send_error.call_count == 1 @@ -93,7 +93,7 @@ def test_execute_throws_if_fatal_error_happens_for_execution(): ) assert execute_cell_mock.return_value is ret assert kernel._fatal_error == message - assert execute_cell_mock.called_once_with("None", True) + assert execute_cell_mock.assert_called_once_with("None", True) assert ipython_display.send_error.call_count == 1