memray_memory_allocation_tracking_with_context_manager.py

1import memray
2
3def memory_intensive_function():
4    return [i for i in range(1000000)]
5
6if __name__ == "__main__":
7    with memray.Tracker("output_file.bin"):
8        print("Starting memory intensive function...")
9        data = memory_intensive_function()
10        print("Finished memory intensive function.")