FIX: finish transform

jklymak · jklymak · commit c31c4f64cef8 · 2019-03-12T10:22:51.000-07:00
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -227,6 +227,9 @@ def get_datalim(self, transData):
                 # (i.e. like scatter). We can't uniquely set limits based on
                 # those shapes, so we just set the limits based on their
                 # location.
+                # Finish the transform:
+                offsets = transOffset.transform(offsets)
+                offsets = transData.inverted().transform(offsets)
                 offsets = np.ma.masked_invalid(offsets)
                 extents = [[np.min(offsets[:, 0]), np.min(offsets[:, 1])],
                            [np.max(offsets[:, 0]), np.max(offsets[:, 1])]]
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -428,11 +428,11 @@ def test_quiver_limits():
 
 def test_barb_limits():
     ax = plt.axes()
-    x = np.linspace(-5, 10, 20) + 25
-    y = np.linspace(-2, 4, 10) + 32
+    x = np.linspace(-5, 10, 20)
+    y = np.linspace(-2, 4, 10)
     y, x = np.meshgrid(y, x)
-    #    trans = mtransforms.Affine2D().translate(25, 32) + ax.transData
-    plt.barbs(x, y, np.sin(x), np.cos(y))
+    trans = mtransforms.Affine2D().translate(25, 32) + ax.transData
+    plt.barbs(x, y, np.sin(x), np.cos(y), transform=trans)
     # The calculated bounds are approximately the bounds of the original data,
     # this is because the entire path is taken into account when updating the
     # datalim.
