[Work][Python]Simple CPU stress script

Key technique: multiprocessing, Pool, Psutil

Task:

Let CPU usage reach to 100%.

Analysis:

Use multi threads to fill every logical processors in system.

Solution:

from multiprocessing import Pool
import psutil
import time
def f(x):
    set_time = 1
    timeout = time.time() + 60*float(set_time)  # X minutes from now
    while True:
        if time.time() > timeout:
            break
if __name__ == '__main__':
    processes = psutil.cpu_count()
    print ('utilizing %d cores\n' % processes)
    pool = Pool(processes)
    pool.map(f, range(processes))

Result:

Reference:

https://gist.github.com/mda590/7a9a6b21b74ae10aa350b1703e2724a0