python 2 unittest doesn't support assertRaisesRegex

nicholas-esterer · nicholas-esterer · commit b8e23c4e4476 · 2020-10-30T16:26:47.000-04:00
diff --git a/packages/python/plotly/plotly/tests/test_core/test_graph_objs/test_figure.py b/packages/python/plotly/plotly/tests/test_core/test_graph_objs/test_figure.py
@@ -215,5 +215,10 @@ def test_set_subplots(self):
         )
         assert fig1.layout == fig1_sp.layout
         # Test that calling on a figure that already has subplots throws an error.
-        with self.assertRaisesRegex(ValueError, "^This figure already has subplots\.$"):
+        raised = False
+        try:
             fig1.set_subplots(2, 3)
+        except ValueError as e:
+            assert e.args[0] == "This figure already has subplots."
+            raised = True
+        assert raised
