Updated tests to check for default value of shading

jeffreypaul15 · jeffreypaul15 · commit 95f40030f147 · 2021-05-29T10:09:11.000+05:30
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -743,7 +743,6 @@ def test_quadmesh_deprecated_signature(
     qmesh = QuadMesh(coords, **kwargs)
     qmesh.set_array(C)
     ax.add_collection(qmesh)
-
     assert qmesh._shading == 'flat'
 
     ax = fig_ref.add_subplot()
@@ -756,6 +755,7 @@ def test_quadmesh_deprecated_signature(
                          **kwargs)
     qmesh.set_array(C.flatten() if flat_ref else C)
     ax.add_collection(qmesh)
+    assert qmesh._shading == 'flat'
 
 
 @check_figures_equal(extensions=['png'])
