Add test for grouped_bar() return value

Author: timhoffm

lib/matplotlib/axes/_axes.py

@@ -78,7 +78,7 @@ def __init__(self, bar_containers):
         self.bar_containers = bar_containers
 
     def remove(self):
-        [b.remove() for b in self.bars]
+        [b.remove() for b in self.bar_containers]
 
 
 @_docstring.interpd

lib/matplotlib/tests/test_axes.py

@@ -23,6 +23,7 @@
 from matplotlib import rc_context, patheffects
 import matplotlib.colors as mcolors
 import matplotlib.dates as mdates
+from matplotlib.container import BarContainer
 from matplotlib.figure import Figure
 from matplotlib.axes import Axes
 from matplotlib.lines import Line2D
@@ -2251,6 +2252,20 @@ def test_grouped_bar_dataframe(fig_test, fig_ref, pd):
     ax.legend()
 
 
+def test_grouped_bar_return_value():
+    fig, ax = plt.subplots()
+    ret = ax.grouped_bar([[1, 2, 3], [11, 12, 13]], tick_labels=['A', 'B', 'C'])
+
+    assert len(ret.bar_containers) == 2
+    for bc in ret.bar_containers:
+        assert isinstance(bc, BarContainer)
+        assert bc in ax.containers
+
+    ret.remove()
+    for bc in ret.bar_containers:
+        assert bc not in ax.containers
+
+
 def test_boxplot_dates_pandas(pd):
     # smoke test for boxplot and dates in pandas
     data = np.random.rand(5, 2)