Add test for grouped_bar() return value

timhoffm · timhoffm · commit d9aa5f64bae9 · 2025-06-02T16:06:33.000+02:00
diff --git a/lib/matplotlib/axes/_axes.py b/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
diff --git a/lib/matplotlib/tests/test_axes.py b/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)
