Measures Time taken and Memory before and after for a block of Python code

measure_time_memory.py

    

import time
import os
import psutil

class MeasureMemTime:
    def __init__(self):
        self.start = None
        self.end = None
        self.startmem = None
        self.endmem = None
 
    def __enter__(self):
        self.start = time.time()
        self.startmem = psutil.Process(os.getpid()).get_memory_info().rss
        print("Memory before start (Bytes): ", self.startmem)
        return self
 
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.end = time.time()
        self.endmem = psutil.Process(os.getpid()).get_memory_info().rss
        print("Total time taken (ms): ", self.duration())
        print("Memory after finish (Bytes): ", self.endmem)
 
    def duration(self):
        return str((self.end - self.start) * 1000) + ' milliseconds'

import time
 
def foo():
    time.sleep(1)
 
with MeasureMemTime() as m:
   foo()    # We can place here the multiple calls or  arbitary code block to measure