moved super.__init()__

jeffreypaul15 · jeffreypaul15 · commit 09f62a4c791a · 2021-05-13T12:09:29.000+05:30
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
@@ -367,6 +367,10 @@ def set_array(self, A):
         ----------
         A : ndarray or None
         """
+        if A is None:
+            self._A = None
+            return
+
         if isinstance(self, mpl.collections.QuadMesh):
             A_ = A.ravel()
             width, height = self._meshWidth, self._meshHeight
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1993,7 +1993,6 @@ class QuadMesh(Collection):
     """
     def __init__(self, meshWidth, meshHeight, coordinates,
                  antialiased=True, shading='flat', **kwargs):
-        super().__init__(**kwargs)
         self._meshWidth = meshWidth
         self._meshHeight = meshHeight
         # By converting to floats now, we can avoid that on every draw.
@@ -2005,6 +2004,7 @@ def __init__(self, meshWidth, meshHeight, coordinates,
         self._bbox = transforms.Bbox.unit()
         self._bbox.update_from_data_xy(coordinates.reshape(
             ((meshWidth + 1) * (meshHeight + 1), 2)))
+        super().__init__(**kwargs)
 
     def get_paths(self):
         if self._paths is None:
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -1,4 +1,5 @@
 import io
+import re
 from types import SimpleNamespace
 
 import numpy as np
@@ -706,17 +707,17 @@ def test_quadmesh_set_array_validation():
     coll = ax.pcolormesh(x, y, np.ones(z.shape))
     z = np.arange(16).reshape((4, 4))
 
-    with pytest.raises(TypeError, match="Dimensions of A (4, 4) "
-                       "are incompatible with X (3) and/or Y (3)"):
+    with pytest.raises(TypeError, match=re.escape("Dimensions of A (4, 4) "
+                       "are incompatible with X (3) and/or Y (3)")):
         coll.set_array(z)
-    with pytest.raises(TypeError, match="Dimensions of A (16,) "
-                       "are incompatible with X (3) and/or Y (3)"):
+    with pytest.raises(TypeError, match=re.escape("Dimensions of A (16,) "
+                       "are incompatible with X (3) and/or Y (3)")):
         coll.set_array(z.ravel())
 
     fig, ax = plt.subplots()
     coll = ax.pcolormesh(x, y, np.ones(z.shape), shading='nearest')
-    with pytest.raises(TypeError, match="Dimensions of A (3, 3) "
-                       "are incompatible with X (4) and/or Y (4)"):
+    with pytest.raises(TypeError, match=re.escape("Dimensions of A (3, 3) "
+                       "are incompatible with X (4) and/or Y (4)")):
         z = np.arange(9).reshape((3, 3))
         coll.set_array(z)
 
@@ -726,8 +727,8 @@ def test_quadmesh_set_array_validation():
     fig, ax = plt.subplots()
     coll = ax.pcolormesh(x, y, np.ones(z.shape), shading='gouraud')
     z = np.arange(16).reshape((4, 4))
-    with pytest.raises(TypeError, match="Dimensions of A (4, 4) "
-                       "are incompatible with X (2) and/or Y (2)"):
+    with pytest.raises(TypeError, match=re.escape("Dimensions of A (4, 4) "
+                       "are incompatible with X (2) and/or Y (2)")):
         coll.set_array(z)
 
 
