test and changelog entry

Author: kevineger

CHANGELOG.md

@@ -5,6 +5,7 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 ## Unreleased
 
 ### Fixed
+- Fix Colab renderer detection to use `COLAB_NOTEBOOK_ID` env var instead of importing `google.colab`, so the `colab` renderer is only set for Colab web notebooks and not for other environments like the Colab VS Code extension [[#5473](https://github.com/plotly/plotly.py/pull/5473)]
 - Fix issue where user-specified `color_continuous_scale` was ignored when template had `autocolorscale=True` [[#5439](https://github.com/plotly/plotly.py/pull/5439)], with thanks to @antonymilne for the contribution!
 - Update tests to be compatible with numpy 2.4 [[#5522](https://github.com/plotly/plotly.py/pull/5522)], with thanks to @thunze for the contribution!
 

tests/test_io/test_renderers.py

@@ -1,4 +1,5 @@
 import json
+import os
 import threading
 import time
 
@@ -418,3 +419,78 @@ def test_missing_webbrowser_methods(fig1):
     finally:
         # restore everything after this test
         webbrowser.get = removed_webbrowser_get_method
+
+
+def test_colab_renderer_when_env_var_is_set():
+    """
+    When COLAB_NOTEBOOK_ID is present the default renderer should be 'colab'.
+    """
+    import importlib
+    import plotly.io._renderers as _renderers_mod
+    from plotly import optional_imports
+
+    fake_ipython = MagicMock()
+    original_get_module = optional_imports.get_module
+
+    def patched_get_module(name, *args, **kwargs):
+        if name in ("IPython", "IPython.display"):
+            return fake_ipython
+        return original_get_module(name, *args, **kwargs)
+
+    original_default = pio.renderers.default
+    try:
+        with mock.patch.dict(os.environ, {"COLAB_NOTEBOOK_ID": "fake-id"}, clear=True):
+            with mock.patch.object(
+                optional_imports, "get_module", side_effect=patched_get_module
+            ):
+                importlib.reload(_renderers_mod)
+                assert _renderers_mod.renderers.default == "colab"
+    finally:
+        importlib.reload(_renderers_mod)
+        pio.renderers.default = original_default
+
+
+def test_colab_renderer_when_env_var_is_absent():
+    """
+    Without COLAB_NOTEBOOK_ID the default renderer must not be 'colab',
+    even when ``google.colab`` is importable.  The old detection used
+    ``import google.colab`` which wrongly activated the colab renderer
+    in environments that merely had the package installed (e.g. the
+    Colab VS Code extension).
+
+    Regression test for https://github.com/plotly/plotly.py/pull/5473.
+    """
+    import sys
+    import types
+    import importlib
+    import plotly.io._renderers as _renderers_mod
+    from plotly import optional_imports
+
+    fake_ipython = MagicMock()
+    original_get_module = optional_imports.get_module
+
+    def patched_get_module(name, *args, **kwargs):
+        if name in ("IPython", "IPython.display"):
+            return fake_ipython
+        return original_get_module(name, *args, **kwargs)
+
+    # Make google.colab importable so the old ``import google.colab``
+    # approach would have chosen the colab renderer here.
+    fake_google = types.ModuleType("google")
+    fake_google_colab = types.ModuleType("google.colab")
+
+    original_default = pio.renderers.default
+    try:
+        with mock.patch.dict(os.environ, {}, clear=True):
+            with mock.patch.dict(
+                sys.modules,
+                {"google": fake_google, "google.colab": fake_google_colab},
+            ):
+                with mock.patch.object(
+                    optional_imports, "get_module", side_effect=patched_get_module
+                ):
+                    importlib.reload(_renderers_mod)
+                    assert _renderers_mod.renderers.default != "colab"
+    finally:
+        importlib.reload(_renderers_mod)
+        pio.renderers.default = original_default