From 9b36630c249f3d776dd90a75bd736949bf2bfd04 Mon Sep 17 00:00:00 2001 From: Devin Stein Date: Tue, 12 Nov 2024 21:27:49 -0700 Subject: [PATCH] fix(tests): fix test logic --- sparkmagic/sparkmagic/tests/test_sparkkernelbase.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py b/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py index 07afca2a..4aeb0f0a 100644 --- a/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py +++ b/sparkmagic/sparkmagic/tests/test_sparkkernelbase.py @@ -45,11 +45,10 @@ def test_execute_valid_code(): ret = kernel.do_execute(code, False) kernel.user_code_parser.get_code_to_run.assert_called_once_with(code) - execute_cell_mock.assert_called_once_with(ret, True) + + execute_cell_mock.assert_called_once_with(code, False, True, None, False) assert execute_cell_mock.return_value is ret assert kernel._fatal_error is None - - execute_cell_mock.assert_called_once_with(code, True) assert ipython_display.send_error.call_count == 0