TST: Capture UserWarning for Bar plot with MultiIndex

nrebena · nrebena · commit 71de5e21d06e · 2020-06-17T21:26:40.000+02:00
diff --git a/pandas/tests/plotting/test_frame.py b/pandas/tests/plotting/test_frame.py
@@ -3422,7 +3422,9 @@ def test_bar_multiindex(self):
         errors = gp3.std()
 
         # No assertion we just ensure that we can plot a MultiIndex bar plot
-        means.plot.bar(yerr=errors, capsize=4)
+        # and are getting a UserWarning
+        with tm.assert_produces_warning(UserWarning):
+            means.plot.bar(yerr=errors, capsize=4)
 
 
 def _generate_4_axes_via_gridspec():
