Merge pull request #162 from romgar/fix_flaky_memory_test

Call gc.collect() before measuring memory usage.
romgar · Jan 3, 2021 · 9c31f45 · 9c31f45
2 parents ee039fe + cf748e2
commit 9c31f45
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_memory_leak.py b/tests/test_memory_leak.py
@@ -1,3 +1,4 @@
+import gc
 import resource
 
 import pytest
@@ -9,8 +10,10 @@
 
 def test_rss_usage():
     DirtyMixinModel()
+    gc.collect()
     rss_1 = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
     for _ in range(1000):
         DirtyMixinModel()
+    gc.collect()
     rss_2 = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
     assert rss_2 == rss_1, 'There is a memory leak!'