From 9a2d54d2c1551605ac666d56311291b7d3fa3ac4 Mon Sep 17 00:00:00 2001
From: Martin Fleischmann <martin@martinfleischmann.net>
Date: Thu, 16 Jan 2025 15:02:41 +0100
Subject: [PATCH] TST: matplotlib test compatibility (#150)

---
 pointpats/tests/test_kde.py | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/pointpats/tests/test_kde.py b/pointpats/tests/test_kde.py
index 8f39af7..f103438 100644
--- a/pointpats/tests/test_kde.py
+++ b/pointpats/tests/test_kde.py
@@ -1,5 +1,6 @@
 import numpy as np
 import pytest
+from packaging.version import Version
 
 from pointpats import plot_density
 
@@ -7,7 +8,10 @@
 statsmodels = pytest.importorskip("statsmodels")
 KDEpy = pytest.importorskip("KDEpy")
 
+MPL_310 = Version(matplotlib.__version__) >= Version("3.10.0")
 
+
+@pytest.mark.skipif(not MPL_310, reason="ContourSet is composed differently in < 3.10")
 class TestDensity:
     def setup_method(self):
         self.points = np.array(
@@ -30,7 +34,7 @@ def setup_method(self):
     def test_default(self):
         ax = plot_density(self.points, 10)
         contourset = ax.collections[0]
-        assert len(contourset.collections) == 12
+        assert len(ax.collections[0].get_paths()) == 12
         assert not contourset.filled
         np.testing.assert_array_equal(contourset.get_linewidths(), np.array([1.5] * 12))
         np.testing.assert_array_equal(
@@ -40,33 +44,28 @@ def test_default(self):
 
     def test_bandwidth(self):
         ax = plot_density(self.points, 1)
-        contourset = ax.collections[0]
-        assert len(contourset.collections) == 10
+        assert len(ax.collections[0].get_paths()) == 10
 
     def test_resolution(self):
         ax = plot_density(self.points, 10, resolution=200)
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 12
+        assert len(contourset.get_paths()) == 12
 
     def test_margin(self):
         ax = plot_density(self.points, 10, margin=0.3)
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 12
+        assert len(contourset.get_paths()) == 12
 
     def test_kdepy(self):
         ax = plot_density(self.points, 10, kernel="gaussian")
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 12
+        assert len(contourset.get_paths()) == 12
         np.testing.assert_array_equal(contourset.get_linewidths(), np.array([1.5] * 12))
 
     def test_levels(self):
         ax = plot_density(self.points, 10, levels=5)
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 7
+        assert len(contourset.get_paths()) == 7
 
     def test_fill(self):
         ax = plot_density(self.points, 10, fill=True)
@@ -84,8 +83,7 @@ def test_geopandas(self):
         gs = geopandas.GeoSeries.from_xy(*self.points.T)
         ax = plot_density(gs, 10)
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 12
+        assert len(contourset.get_paths()) == 12
         np.testing.assert_array_equal(contourset.get_linewidths(), np.array([1.5] * 12))
 
     def test_kwargs(self):
@@ -93,8 +91,7 @@ def test_kwargs(self):
             self.points, 10, cmap="magma", linewidths=0.5, linestyles="-."
         )
         contourset = ax.collections[0]
-        collections = contourset.collections
-        assert len(collections) == 12
+        assert len(contourset.get_paths()) == 12
         np.testing.assert_array_equal(contourset.get_linewidths(), np.array([0.5] * 12))
 
         np.testing.assert_array_equal(