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 from __future__ import print_function
   6 import arvados
   7 import sys
   8 import os
   9
  10 if "JOB_UUID" in os.environ:
  11     requested = arvados.api().jobs().get(uuid=os.environ["JOB_UUID"]).execute()["runtime_constraints"]["min_ram_mb_per_node"]
  12 else:
  13     requested = arvados.api().containers().current().execute()["runtime_constraints"]["ram"]/(1024*1024)
  14
  15 print("Requested %d expected %d" % (requested, int(sys.argv[1])))
  16
  17 exit(0 if requested == int(sys.argv[1]) else 1)