move extra gc into clean func.

pygfx · Feb 7, 2024 · 567f498 · 567f498
1 parent 65044aa
commit 567f498
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests_mem/testutils.py b/tests_mem/testutils.py
@@ -81,8 +81,9 @@ def clear_mem():
     if is_pypy:
         gc.collect()
 
-    device = wgpu.utils.get_default_device()
-    device._poll()
+    wgpu.utils.get_default_device()._poll()
+
+    gc.collect()
 
 
 def get_counts():
@@ -190,9 +191,7 @@ def core_test_func():
             assert ob_name == cls.__name__[3:]
 
             # Give wgpu some slack to clean up temporary resources
-            gc.collect()  # removes the additional CommandEncoder and Textures
-            wgpu.utils.get_default_device()._poll()
-            gc.collect()  # removes the additional CommandBuffer
+            clear_mem()
 
             # Measure peak object counts
             counts2 = get_counts()