sdk/cwl/tests/wf/check_mem.py

   1 import arvados
   2 import sys
   3 import os
   4
   5 if "JOB_UUID" in os.environ:
   6     requested = arvados.api().jobs().get(uuid=os.environ["JOB_UUID"]).execute()["runtime_constraints"]["min_ram_mb_per_node"]
   7 else:
   8     requested = arvados.api().containers().current().execute()["runtime_constraints"]["ram"]/(1024*1024)
   9
  10 print("Requested %d expected %d" % (requested, int(sys.argv[1])))
  11
  12 exit(0 if requested == int(sys.argv[1]) else 1)