codestyle

jeffreypaul15 · jeffreypaul15 · commit 09ca6faa9fdd · 2021-05-12T23:00:25.000+05:30
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -706,14 +706,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(ValueError, match="Dimensions of A (4, 4) are incompatible with X (3) and/or Y (3)"):
+    with pytest.raises(ValueError, match="Dimensions of A (4, 4) "
+                       "are incompatible with X (3) and/or Y (3)"):
         coll.set_array(z)
-    with pytest.raises(ValueError, match="Dimensions of A (16,) are incompatible with X (3) and/or Y (3)"):
+    with pytest.raises(ValueError, match="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(ValueError, match="Dimensions of A (3, 3) are incompatible with X (4) and/or Y (4)"):
+    with pytest.raises(ValueError, match="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)
 
@@ -723,7 +726,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(ValueError, match="Dimensions of A (4, 4) are incompatible with X (2) and/or Y (2)"):
+    with pytest.raises(ValueError, match="Dimensions of A (4, 4) "
+                       "are incompatible with X (2) and/or Y (2)"):
         coll.set_array(z)
 
 
