FIX: Testing on Pypy (#353)

pygeos · Jul 5, 2021 · e906ce9 · e906ce9
1 parent 97e455d
commit e906ce9
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pygeos/test/common.py b/pygeos/test/common.py
@@ -2,6 +2,7 @@
 from contextlib import contextmanager
 
 import numpy as np
+import pytest
 
 import pygeos
 
@@ -52,14 +53,20 @@
 
 @contextmanager
 def assert_increases_refcount(obj):
-    before = sys.getrefcount(obj)
+    try:
+        before = sys.getrefcount(obj)
+    except AttributeError:  # happens on Pypy
+        pytest.skip("sys.getrefcount is not available.")
     yield
     assert sys.getrefcount(obj) == before + 1
 
 
 @contextmanager
 def assert_decreases_refcount(obj):
-    before = sys.getrefcount(obj)
+    try:
+        before = sys.getrefcount(obj)
+    except AttributeError:  # happens on Pypy
+        pytest.skip("sys.getrefcount is not available.")
     yield
     assert sys.getrefcount(obj) == before - 1