tests for deprecation warnings in graph_objects

emilykl · emilykl · commit 553d6ac2042c · 2024-12-02T15:18:50.000-05:00
diff --git a/packages/python/plotly/plotly/tests/test_core/test_graph_objs/test_graph_objs.py b/packages/python/plotly/plotly/tests/test_core/test_graph_objs/test_graph_objs.py
@@ -1,4 +1,5 @@
 from unittest import TestCase
+import warnings
 
 import plotly.graph_objs as go
 
@@ -154,3 +155,29 @@ def test_pop_invalid_prop_key_error(self):
 
     def test_pop_invalid_prop_with_default(self):
         self.assertEqual(self.layout.pop("bogus", 42), 42)
+
+
+class TestDeprecationWarnings(TestCase):
+    def test_warn_on_deprecated_mapbox_traces(self):
+        # This test will fail if any of the following traces
+        # fails to emit a warning
+        for trace_constructor in [
+            go.Scattermapbox,
+            go.Densitymapbox,
+            go.Choroplethmapbox,
+        ]:
+            with self.assertWarns(DeprecationWarning):
+                _ = go.Figure([trace_constructor()])
+
+    def test_no_warn_on_other_traces(self):
+        # This test will fail if any of the following traces emits a warning
+        for trace_constructor in [
+            go.Scatter,
+            go.Bar,
+            go.Scattermap,
+            go.Densitymap,
+            go.Choroplethmap,
+        ]:
+            with warnings.catch_warnings():
+                warnings.simplefilter("error")
+                _ = go.Figure([trace_constructor()])
