From b2d746be07c185c828aca386fae8b1fddb7d2980 Mon Sep 17 00:00:00 2001 From: Elliott Sales de Andrade Date: Fri, 25 Apr 2025 23:10:55 -0400 Subject: [PATCH] TST: Make refcount tests more resilient to Python changes Check the change of the refcount, instead of the absolute value, as suggested by @ngoldbaum. --- lib/matplotlib/tests/test_quiver.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/lib/matplotlib/tests/test_quiver.py b/lib/matplotlib/tests/test_quiver.py index cd1e1c3c7042..1205487cfe94 100644 --- a/lib/matplotlib/tests/test_quiver.py +++ b/lib/matplotlib/tests/test_quiver.py @@ -26,11 +26,12 @@ def test_quiver_memory_leak(): Q = draw_quiver(ax) ttX = Q.X + orig_refcount = sys.getrefcount(ttX) Q.remove() del Q - assert sys.getrefcount(ttX) == 2 + assert sys.getrefcount(ttX) < orig_refcount @pytest.mark.skipif(platform.python_implementation() != 'CPython', @@ -43,9 +44,9 @@ def test_quiver_key_memory_leak(): qk = ax.quiverkey(Q, 0.5, 0.92, 2, r'$2 \frac{m}{s}$', labelpos='W', fontproperties={'weight': 'bold'}) - assert sys.getrefcount(qk) == 3 + orig_refcount = sys.getrefcount(qk) qk.remove() - assert sys.getrefcount(qk) == 2 + assert sys.getrefcount(qk) < orig_refcount def test_quiver_number_of_args():