test: fix IPython import mocking in Jupyter stderr tests

- Mock builtins.__import__ instead of patching IPython.display directly

- Handle 'from IPython.display import HTML, display' import pattern

- Tests now work without IPython installed in test environment

Author: dgenio

tests/client/test_stdio.py

@@ -726,22 +726,68 @@ def test_print_stderr_non_jupyter():
 def test_print_stderr_jupyter():
     """Test stderr printing when in Jupyter using IPython display."""
     # Mock the Jupyter detection and IPython display
+    # We need to mock the import inside the function since IPython may not be installed
+    mock_html_class = MagicMock()
+    mock_display_func = MagicMock()
+    
+    # Create a mock module structure that matches "from IPython.display import HTML, display"
+    mock_display_module = MagicMock()
+    mock_display_module.HTML = mock_html_class
+    mock_display_module.display = mock_display_func
+    
+    # Create mock IPython module with display submodule
+    mock_ipython_module = MagicMock()
+    mock_ipython_module.display = mock_display_module
+    
+    original_import = __import__
+    
+    def mock_import(name, globals=None, locals=None, fromlist=(), level=0):
+        if name == "IPython.display":
+            return mock_display_module
+        if name == "IPython":
+            return mock_ipython_module
+        # For other imports, use real import
+        return original_import(name, globals, locals, fromlist, level)
+    
     with patch("mcp.client.stdio._is_jupyter_notebook", return_value=True), patch(
-        "IPython.display.display"
-    ) as mock_display, patch("IPython.display.HTML") as mock_html:
+        "builtins.__import__", side_effect=mock_import
+    ):
         _print_stderr("test error message", sys.stderr)
 
         # Verify IPython display was called
-        mock_html.assert_called_once()
-        mock_display.assert_called_once()
+        mock_html_class.assert_called_once()
+        mock_display_func.assert_called_once()
 
 
 def test_print_stderr_jupyter_fallback():
     """Test stderr printing falls back to regular print if IPython display fails."""
     stderr_capture = io.StringIO()
+    
+    # Mock IPython import to raise exception on display
+    mock_html_class = MagicMock()
+    mock_display_func = MagicMock(side_effect=Exception("Display failed"))
+    
+    # Create a mock module structure that matches "from IPython.display import HTML, display"
+    mock_display_module = MagicMock()
+    mock_display_module.HTML = mock_html_class
+    mock_display_module.display = mock_display_func
+    
+    # Create mock IPython module with display submodule
+    mock_ipython_module = MagicMock()
+    mock_ipython_module.display = mock_display_module
+    
+    original_import = __import__
+    
+    def mock_import(name, globals=None, locals=None, fromlist=(), level=0):
+        if name == "IPython.display":
+            return mock_display_module
+        if name == "IPython":
+            return mock_ipython_module
+        # For other imports, use real import
+        return original_import(name, globals, locals, fromlist, level)
 
     with patch("mcp.client.stdio._is_jupyter_notebook", return_value=True), patch(
-        "IPython.display.display", side_effect=Exception("Display failed")
+        "builtins.__import__", side_effect=mock_import
     ):
         _print_stderr("test error message", stderr_capture)