test(display): catch expected UserWarning in test_repr_mimebundle_selection_logic

Author: shuoweil

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

@@ -14,6 +14,8 @@
 
 import unittest.mock as mock
 
+import pytest
+
 import bigframes.display.html as bf_html
 import bigframes.pandas as bpd
 
@@ -63,7 +65,8 @@ def test_repr_mimebundle_selection_logic():
         # Test fallback to static deferred repr when anywidget fails
         mock_anywidget.side_effect = Exception("Anywidget failed")
         with bpd.option_context("display.repr_mode", "deferred"):
-            bundle = bf_html.repr_mimebundle(mock_obj)
+            with pytest.warns(UserWarning, match="Anywidget mode is not available"):
+                bundle = bf_html.repr_mimebundle(mock_obj)
             assert bundle == {"text/plain": "deferred"}
             mock_deferred.assert_called_once()