FIX: Handle height units in brohen_barh()

Author: timhoffm

lib/matplotlib/axes/_axes.py

@@ -2243,13 +2243,10 @@ def _convert_dx(dx, x0, xconv, convert):
             except (TypeError, IndexError, KeyError):
                 x = xconv
 
-            delist = False
-            if not np.iterable(dx):
-                dx = [dx]
-                delist = True
-            dx = [convert(x0 + ddx) - x for ddx in dx]
-            if delist:
-                dx = dx[0]
+            if np.iterable(dx):
+                dx = [convert(x0 + ddx) - x for ddx in dx]
+            else:
+                dx = convert(x0 + dx) - x
         except (ValueError, TypeError, AttributeError):
             # if the above fails (for any reason) just fallback to what
             # we do by default and convert dx by itself.
@@ -3037,15 +3034,21 @@ def broken_barh(self, xranges, yrange, align="bottom", **kwargs):
             [("x", xdata), ("y", ydata)], kwargs, convert=False)
 
         vertices = []
-        y0, dy = yrange
+        ypos, height = yrange
+
+        # Unit conversion: handling of the difference quantity height is done through
+        # _convert_dx() in the same way as width handling in bar().
+        y0 = self.convert_yunits(ypos)
+        dy = self._convert_dx(height, ypos, np.array(y0), self.convert_yunits)
+        print(type(y0), type(dy))
 
         _api.check_in_list(['bottom', 'center', 'top'], align=align)
         if align == "bottom":
-            y0, y1 = self.convert_yunits((y0, y0 + dy))
+            y1 = y0 + dy
         elif align == "center":
-            y0, y1 = self.convert_yunits((y0 - dy/2, y0 + dy/2))
+            y0, y1 = y0 - dy / 2, y0 + dy / 2
         else:
-            y0, y1 = self.convert_yunits((y0 - dy, y0))
+            y0, y1 = y0 - dy, y0
 
         for xr in xranges:  # convert the absolute values, not the x and dx
             try:
@@ -3057,6 +3060,7 @@ def broken_barh(self, xranges, yrange, align="bottom", **kwargs):
             x0, x1 = self.convert_xunits((x0, x0 + dx))
             vertices.append([(x0, y0), (x0, y1), (x1, y1), (x1, y0)])
 
+        print(vertices)
         col = mcoll.PolyCollection(np.array(vertices), **kwargs)
         self.add_collection(col)
 

lib/matplotlib/tests/test_axes.py

@@ -7748,6 +7748,18 @@ def test_broken_barh_align():
         assert_array_equal(path.get_extents().intervaly, [18, 20])
 
 
+def test_broken_barh_categorical():
+    fig, ax = plt.subplots()
+    pc = ax.broken_barh([(0, 10)], ('a', 0.8))
+    assert tuple(pc.get_datalim(ax.transData).intervaly) == (0, 0.8)
+
+    pc = ax.broken_barh([(0, 10)], ('a', 0.8), align="center")
+    assert tuple(pc.get_datalim(ax.transData).intervaly) == (-0.4, 0.4)
+
+    pc = ax.broken_barh([(0, 10)], ('a', 0.8), align="top")
+    assert tuple(pc.get_datalim(ax.transData).intervaly) == (-0.8, 0)
+
+
 def test_pandas_pcolormesh(pd):
     time = pd.date_range('2000-01-01', periods=10)
     depth = np.arange(20)