# Wersja z generatorem

#!/usr/bin/env python
from memory_profiler import memory_usage

def get_data():
    for i in range(0, 10000000):
        yield i

def fun():
    for i in get_data():
        pass

mem = max(memory_usage(proc=fun))
print("Max memory:", mem, "MiB")

# Wersja z listą

#!/usr/bin/env python
from memory_profiler import memory_usage

def get_data():
    return [i for i in range(0, 10000000)]

def fun():
    for i in get_data():
        pass

mem = max(memory_usage(proc=fun))
print("Max memory:", mem, "MiB")