test: mock get_ipython_execution_count directly in unit tests

Author: shuoweil

packages/bigframes/tests/unit/display/test_anywidget.py

@@ -180,14 +180,10 @@ def test_page_size_change_resets_sort(mock_df):
 def test_cell_execution_count_propagation(mock_df):
     """Test that the captured cell_execution_count is propagated to to_pandas_batches."""
     with mock.patch(
-        "IPython.core.interactiveshell.InteractiveShell.initialized", return_value=True
+        "bigframes.core.utils.get_ipython_execution_count", return_value=42
     ):
-        with mock.patch(
-            "IPython.core.interactiveshell.InteractiveShell.instance"
-        ) as mock_instance:
-            mock_instance.return_value.execution_count = 42
-            with bigframes.option_context("display.render_mode", "anywidget"):
-                widget = TableWidget(mock_df)
+        with bigframes.option_context("display.render_mode", "anywidget"):
+            widget = TableWidget(mock_df)
 
     assert widget._cell_execution_count == 42
 

packages/bigframes/tests/unit/session/test_read_gbq_colab.py

@@ -208,13 +208,9 @@ def test_execution_history_filters_by_notebook_cell_when_all_cells_is_false():
     session._metrics.jobs.extend([job1, job2])
 
     with mock.patch(
-        "IPython.core.interactiveshell.InteractiveShell.initialized", return_value=True
+        "bigframes.core.utils.get_ipython_execution_count", return_value=20
     ):
-        with mock.patch(
-            "IPython.core.interactiveshell.InteractiveShell.instance"
-        ) as mock_instance:
-            mock_instance.return_value.execution_count = 20
-            history = session.execution_history(all_cells=False).to_dataframe()
+        history = session.execution_history(all_cells=False).to_dataframe()
 
     assert len(history) == 1
     assert history.iloc[0]["job_id"] == "job_2"