@@ -86,7 +86,7 @@ def _use_gb_ram(num_gb):
86
86
# Build thread list
87
87
thread_list = []
88
88
for idx in range (num_threads ):
89
- thread = Process (target = _use_gb_ram , args = (num_gb / num_threads ,),
89
+ thread = Thread (target = _use_gb_ram , args = (num_gb / num_threads ,),
90
90
name = str (idx ))
91
91
thread_list .append (thread )
92
92
@@ -130,11 +130,11 @@ def setUp(self):
130
130
131
131
# Init parameters
132
132
# Input RAM GB to occupy
133
- self .num_gb = 1.0
133
+ self .num_gb = 0.5
134
134
# Input number of sub-threads (not including parent threads)
135
135
self .num_threads = 2
136
136
# Acceptable percent error for memory profiled against input
137
- self .mem_err_percent = 25
137
+ self .mem_err_gb = 0. 25
138
138
139
139
# ! Only used for benchmarking the profiler over a range of
140
140
# ! RAM usage and number of threads
@@ -380,14 +380,14 @@ def test_cmdline_profiling(self):
380
380
runtime_threads = int (node_stats ['runtime_threads' ])
381
381
382
382
# Get margin of error for RAM GB
383
- allowed_gb_err = ( self .mem_err_percent / 100.0 ) * num_gb
383
+ allowed_gb_err = self .mem_err_gb
384
384
runtime_gb_err = np .abs (runtime_gb - num_gb )
385
385
#
386
386
expected_runtime_threads = num_threads
387
387
388
388
# Error message formatting
389
- mem_err = 'Input memory: %f is not within %.1f%% of runtime ' \
390
- 'memory: %f' % (num_gb , self .mem_err_percent , runtime_gb )
389
+ mem_err = 'Input memory: %f is not within %.3f GB of runtime ' \
390
+ 'memory: %f' % (num_gb , self .mem_err_gb , runtime_gb )
391
391
threads_err = 'Input threads: %d is not equal to runtime threads: %d' \
392
392
% (expected_runtime_threads , runtime_threads )
393
393
@@ -421,14 +421,14 @@ def test_function_profiling(self):
421
421
runtime_threads = int (node_stats ['runtime_threads' ])
422
422
423
423
# Get margin of error for RAM GB
424
- allowed_gb_err = ( self .mem_err_percent / 100.0 ) * num_gb
424
+ allowed_gb_err = self .mem_err_gb
425
425
runtime_gb_err = np .abs (runtime_gb - num_gb )
426
426
#
427
427
expected_runtime_threads = num_threads
428
428
429
429
# Error message formatting
430
- mem_err = 'Input memory: %f is not within %.1f%% of runtime ' \
431
- 'memory: %f' % (num_gb , self .mem_err_percent , runtime_gb )
430
+ mem_err = 'Input memory: %f is not within %.3f GB of runtime ' \
431
+ 'memory: %f' % (num_gb , self .mem_err_gb , runtime_gb )
432
432
threads_err = 'Input threads: %d is not equal to runtime threads: %d' \
433
433
% (expected_runtime_threads , runtime_threads )
434
434
0 commit comments