dont crash on pypy

ipython · Feb 19, 2025 · 6fcdbfe · 6fcdbfe
1 parent 0139eb7
commit 6fcdbfe
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -22,11 +22,16 @@
 
 @pytest.fixture(scope="module", autouse=True)
 def _enable_tracemalloc():
-    import tracemalloc
-
-    tracemalloc.start()
+    try:
+        import tracemalloc
+    except ModuleNotFoundError:
+        # pypy
+        tracemalloc = None
+    if tracemalloc is not None:
+        tracemalloc.start()
     yield
-    tracemalloc.stop()
+    if tracemalloc is not None:
+        tracemalloc.stop()
 
 
 sample_info: dict = {