Update contourf call check for mpl 3.8

pydata · Sep 15, 2023 · c7391fc · c7391fc
1 parent c537238
commit c7391fc
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/xarray/tests/test_plot.py b/xarray/tests/test_plot.py
@@ -167,7 +167,14 @@ def imshow_called(self, plotmethod):
 
     def contourf_called(self, plotmethod):
         plotmethod()
-        paths = plt.gca().findobj(mpl.collections.PathCollection)
+
+        # Compatible with mpl before (PathCollection) and after (QuadContourSet) 3.8
+        def matchfunc(x):
+            return isinstance(
+                x, (mpl.collections.PathCollection, mpl.contour.QuadContourSet)
+            )
+
+        paths = plt.gca().findobj(matchfunc)
         return len(paths) > 0