GH-35728: [CI][Python] Move test_total_bytes_allocated to a subprocess to improve reliability

raulcd · raulcd · commit 0806a80eabd0 · 2023-06-28T15:47:39.000+02:00
diff --git a/python/pyarrow/tests/test_array.py b/python/pyarrow/tests/test_array.py
@@ -24,6 +24,7 @@
 import pickle
 import pytest
 import struct
+import subprocess
 import sys
 import weakref
 
@@ -38,7 +39,17 @@
 
 
 def test_total_bytes_allocated():
+    code = f"""if 1:
+    import pyarrow as pa
+
     assert pa.total_allocated_bytes() == 0
+    """
+    res = subprocess.run([sys.executable, "-c", code],
+                         universal_newlines=True, stderr=subprocess.PIPE)
+    if res.returncode != 0:
+        print(res.stderr, file=sys.stderr)
+        res.check_returncode()  # fail
+    assert len(res.stderr.splitlines()) == 0
 
 
 def test_weakref():
