Allocate n mb per second

mem.py

#!/usr/bin/env python3
from time import sleep
from sys import exit, argv, getsizeof
# this simple python script allocates memory, waits for a while, and then exits successfully
mem = 0
allocs = []
try:
    mem = int(argv[1])
except ValueError:
    print("Usage: mem.py <mb-to-allocate-per-cycle>")
    exit()

bytes = mem * 1024 * 1024
print("Allocating {} bytes".format(bytes), flush=True)
while True:
  sleep(1)

  allocs.append(bytearray(bytes))
  total_size = 0
  for e in allocs:
      total_size = total_size + getsizeof(e)

  print("Allocated size: {}...".format(total_size), flush=True)