Changeset 991df6a in flex_extract.git for python/testsuite.py

Timestamp:

May 14, 2018, 10:11:29 PM (6 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

812283d

Parents:

efdb01a

Message:

finished documentation (except plot_retrieved)

File:

• python/testsuite.py (modified) (2 diffs)

python/testsuite.py

@@ -1 +1 @@
 #!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#************************************************************************
+# TODO AP
+#
+# - provide more tests
+# - provide more documentation
+# -
+#************************************************************************
 
-# This software is licensed under the terms of the Apache Licence Version 2.0
-# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+#*******************************************************************************
+# @Author: Leopold Haimberger (University of Vienna)
 #
-# Leopold Haimberger, Dec 2015
+# @Date: December 2015
 #
-# Functionality provided: This script triggers the ECMWFDATA test suite. Call with
-# testsuite.py [test group]
+# @Change History:
 #
+#    February 2018 - Anne Philipp (University of Vienna):
+#        - applied PEP8 style guide
+#        - added documentation
 #
-# Further documentation may be obtained from www.flexpart.eu
+# @License:
+#    (C) Copyright 2015-2018.
 #
-# Test groups are specified in testsuite.json
-# in addition to a standard python 2.6 or 2.7 installation the following packages need to be installed
-# ECMWF WebMARS, gribAPI with python enabled, emoslib, ecaccess web toolkit, all available from https://software.ecmwf.int/
-# dateutils
-# matplotlib (optional, for debugging)
+#    This software is licensed under the terms of the Apache Licence Version 2.0
+#    which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+#
+# @Program Functionality:
+#    This script triggers the ECMWFDATA test suite. Call with
+#    testsuite.py [test group]
+#
+# @Program Content:
+#
+#*******************************************************************************
 
-import os,sys
+# ------------------------------------------------------------------------------
+# MODULES
+# ------------------------------------------------------------------------------
+import os
+import sys
 import json
 import subprocess
 
+# ------------------------------------------------------------------------------
+# PROGRAM
+# ------------------------------------------------------------------------------
 try:
-    taskfile=open('testsuite.json')
+    taskfile = open('testsuite.json')
 except:
     print 'could not open suite definition file testsuite.json'
@@ -33 +56 @@
     exit()
 
-fprs=os.getenv('FLEXPART_ROOT_SCRIPTS')
+fprs = os.getenv('FLEXPART_ROOT_SCRIPTS')
 if fprs is None:
     print 'FLEXPART_ROOT_SCRIPTS not set .. some test jobs may fail'
 
-tasks=json.load(taskfile,encoding='latin-1')
+tasks = json.load(taskfile, encoding='latin-1')
 taskfile.close()
 if not os.path.exists('../test'):
     os.makedirs('../test')
-if len(sys.argv)>1:
-    groups=sys.argv[1:]
+if len(sys.argv) > 1:
+    groups = sys.argv[1:]
 else:
-    groups=['xinstall','default','ops','work','cv','fc']#,'hires']
-jobcounter=0
-jobfailed=0
+    groups = ['xinstall', 'default', 'ops', 'work', 'cv', 'fc']#,'hires']
+jobcounter = 0
+jobfailed = 0
 for g in groups:
     try:
-        tk,tv=g,tasks[g]
+        tk, tv = g, tasks[g]
     except:
         continue
-    garglist=[]
-    for ttk,ttv in tv.iteritems():
-        if isinstance(ttv,basestring):
-            if ttk!='script':
-                garglist.append('--'+ttk)
-                if '$'==ttv[0]:
+    garglist = []
+    for ttk, ttv in tv.iteritems():
+        if isinstance(ttv, basestring):
+            if ttk != 'script':
+                garglist.append('--' + ttk)
+                if '$' == ttv[0]:
                     garglist.append(os.path.expandvars(ttv))
                 else:
                     garglist.append(ttv)
-    for ttk,ttv in tv.iteritems():
-        if isinstance(ttv,dict):
-            arglist=[]
-            for tttk,tttv in ttv.iteritems():
-                if isinstance(tttv,basestring):
-                        arglist.append('--'+tttk)
+    for ttk, ttv in tv.iteritems():
+        if isinstance(ttv, dict):
+            arglist = []
+            for tttk, tttv in ttv.iteritems():
+                if isinstance(tttv, basestring):
+                        arglist.append('--' + tttk)
                         if '$' in tttv[0]:
                             arglist.append(os.path.expandvars(tttv))
                         else:
                             arglist.append(tttv)
-            print 'Command: ',' '.join([tv['script']]+garglist+arglist)
-            o='../test/'+tk+'_'+ttk+'_'+'_'.join(ttv.keys())
-            print 'Output will be sent to ',o
-            f=open(o,'w')
+            print 'Command: ', ' '.join([tv['script']] + garglist + arglist)
+            o = '../test/' + tk + '_' + ttk + '_' + '_'.join(ttv.keys())
+            print 'Output will be sent to ', o
+            f = open(o, 'w')
             try:
-                p=subprocess.check_call([tv['script']]+garglist+arglist,stdout=f,stderr=f)
+                p = subprocess.check_call([tv['script']] + garglist + arglist,
+                                          stdout=f, stderr=f)
             except:
                 f.write('\nFAILED\n')
                 print 'FAILED'
-                jobfailed+=1
-            jobcounter+=1
+                jobfailed += 1
+            jobcounter += 1
             f.close()
 
 print 'Test suite tasks completed'
-print str(jobcounter-jobfailed)+' successful, '+str(jobfailed)+' failed'
+print str(jobcounter-jobfailed) + ' successful, ' + str(jobfailed) + ' failed'
 print 'If tasks have been submitted via ECACCESS please check emails'