diff --git a/tests/test_connect.py b/tests/test_connect.py index 0d4216bd..c4302387 100644 --- 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 = {