Fix Windows

psf · Jun 12, 2024 · ad789ee · ad789ee
1 parent c0ed773
commit ad789ee
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pyperf/_process_time.py b/pyperf/_process_time.py
@@ -52,6 +52,11 @@ def merge_profile_stats_files(src, dst):
 
 
 def bench_process(loops, args, kw, profile_filename=None):
+    if resource is not None:
+        rusage_children = resource.RUSAGE_CHILDREN
+    else:
+        rusage_children = 0
+
     max_rss = 0
     range_it = range(loops)
     start_time = time.perf_counter()
@@ -61,7 +66,7 @@ def bench_process(loops, args, kw, profile_filename=None):
         args = [args[0], "-m", "cProfile", "-o", temp_profile_filename] + args[1:]
 
     for _ in range_it:
-        start_rss = get_max_rss(resource.RUSAGE_CHILDREN)
+        start_rss = get_max_rss(rusage_children)
 
         proc = subprocess.Popen(args, **kw)
         with proc:
@@ -75,7 +80,7 @@ def bench_process(loops, args, kw, profile_filename=None):
                 os.unlink(temp_profile_filename)
             sys.exit(exitcode)
 
-        rss = get_max_rss(resource.RUSAGE_CHILDREN) - start_rss
+        rss = get_max_rss(rusage_children) - start_rss
         max_rss = max(max_rss, rss)
 
         if profile_filename: