sdk/cwl/tests/wf/check_mem.py

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: Apache-2.0
   4
   5 import arvados
   6 import sys
   7 import os
   8
   9 if "JOB_UUID" in os.environ:
  10     requested = arvados.api().jobs().get(uuid=os.environ["JOB_UUID"]).execute()["runtime_constraints"]["min_ram_mb_per_node"]
  11 else:
  12     requested = arvados.api().containers().current().execute()["runtime_constraints"]["ram"]/(1024*1024)
  13
  14 print("Requested %d expected %d" % (requested, int(sys.argv[1])))
  15
  16 exit(0 if requested == int(sys.argv[1]) else 1)