BUG : fix non-uniform grids in pcolorfast

tacaswell · tacaswell · commit 613ea28017f6 · 2015-04-19T13:09:34.000-04:00
Closes #4227
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -853,8 +853,9 @@ def make_image(self, magnification=1.0):
         l, b, r, t = self.axes.bbox.extents
         width = (round(r) + 0.5) - (round(l) - 0.5)
         height = (round(t) + 0.5) - (round(b) - 0.5)
-        width = width * magnification
-        height = height * magnification
+        # The extra cast-to-int is only needed for python2
+        width = int(round(width * magnification))
+        height = int(round(height * magnification))
         if self._rgbacache is None:
             A = self.to_rgba(self._A, bytes=True)
             self._rgbacache = A
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -3668,6 +3668,17 @@ def test_bar_negative_width():
         assert_equal(b._height, indx + 1)
 
 
+@cleanup
+def test_pcolor_fast_non_uniform():
+    Z = np.arange(6).reshape((3, 2))
+    X = np.array([0, 1, 2, 10])
+    Y = np.array([0, 1, 2])
+
+    plt.figure()
+    ax = plt.subplot(111)
+    ax.pcolorfast(X, Y, Z.T)
+
+
 if __name__ == '__main__':
     import nose
     import sys
