Add a level module fixture that enables tracemalloc

ipython · Feb 19, 2025 · 0139eb7 · 0139eb7
1 parent 087e7aa
commit 0139eb7
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -19,6 +19,16 @@
 
 from .utils import TemporaryWorkingDirectory
 
+
+@pytest.fixture(scope="module", autouse=True)
+def _enable_tracemalloc():
+    import tracemalloc
+
+    tracemalloc.start()
+    yield
+    tracemalloc.stop()
+
+
 sample_info: dict = {
     "ip": "1.2.3.4",
     "transport": "ipc",