testMadrigalBuild.py
111 lines
| 3.5 KiB
| text/x-python
|
PythonLexer
| r0 | import os,os.path,sys | |||
| import traceback | ||||
| """testMadrigalBuild.py verifies whether Madrigal has been successfully built | ||||
| Usage: python testMadrigalBuild.py [MADROOT] | ||||
| If optional argument MADROOT given, search there for test Madrigal file to plot | ||||
| $Id: testMadrigalBuild.py 3304 2011-01-17 15:25:59Z brideout $ | ||||
| """ | ||||
| print """This script will attempt to plot some Madrigal test data. | ||||
| If you see a Pcolor plot that shows electron density versus | ||||
| altitude, the script succeeded. Close the display window | ||||
| when you're done examining it. This may take a few minutes.\n""" | ||||
| import warnings | ||||
| warnings.filterwarnings('ignore', '.*', UserWarning) | ||||
| try: | ||||
| import numpy | ||||
| import matplotlib.pylab | ||||
| import madrigal | ||||
| import madrigal._Madrec | ||||
| import madrigal.ui.madrigalPlot | ||||
| except: | ||||
| traceback.print_exc() | ||||
| print 'FAILURE: REQUIRED MADRIGAL MODULES NOT INSTALLED' | ||||
| sys.exit(-1) | ||||
| if len(sys.argv) > 2: | ||||
| print 'FAILURE: Illegal usage %s' % (str(sys.argv)) | ||||
| sys.exit(-1) | ||||
| if len(sys.argv) == 2: | ||||
| madrootDir = sys.argv[1] | ||||
| if not os.access(os.path.join(madrootDir, 'experiments'), os.R_OK): | ||||
| print 'FAILURE: No experiments directory found in %s' % (madrootDir) | ||||
| sys.exit(-1) | ||||
| else: | ||||
| madDB = madrigal.metadata.MadrigalDB() | ||||
| madrootDir = madDB.getMadroot() | ||||
| # get filename to read | ||||
| filename = os.path.join(madrootDir, 'experiments/1998/mlh/20jan98/mil980120g.003') | ||||
| if not os.access(filename, os.R_OK): | ||||
| filename = os.path.join(madrootDir, 'experiments/1998/mlh/20jan98/mlh980120g.001') | ||||
| if not os.access(filename, os.R_OK): | ||||
| print 'FAILURE: Unable to find basic Madrigal test files installed in experiments/1998/mlh/20jan98' | ||||
| sys.exit(-1) | ||||
| # check that tmpfile can be created | ||||
| if not os.access('/tmp/', os.W_OK): | ||||
| print 'FAILURE: test cannot write to /tmp directory' | ||||
| sys.exit(-1) | ||||
| tmpfile = '/tmp/isprintTest.txt' | ||||
| tmpplot = '/tmp/isprint.png' | ||||
| try: | ||||
| madrigal._Madrec.getIsprintReport(filename, | ||||
| '', | ||||
| ['uth','gdalt','nel'], | ||||
| ['1'], | ||||
| ['uth'], | ||||
| [None], | ||||
| [1], | ||||
| [0.0], | ||||
| [24.0], | ||||
| 0, | ||||
| 0, | ||||
| 0, | ||||
| 'missing', | ||||
| 'assumed', | ||||
| 'knownBad', | ||||
| tmpfile) | ||||
| except: | ||||
| traceback.print_exc() | ||||
| print 'FAILURE: isprint failed' | ||||
| sys.exit(-1) | ||||
| # read isprint string | ||||
| f = open(tmpfile) | ||||
| isprintText = f.read() | ||||
| f.close() | ||||
| os.remove(tmpfile) | ||||
| # produce pcolor plot | ||||
| try: | ||||
| madrigal.ui.madrigalPlot.madPcolorPlot(isprintText, | ||||
| 'Nel (log(m^-3)) - Millstone Hill - Test day Jan 20, 1998\nClose this Window if okay', | ||||
| 'Hours since midnight UT Jan 20, 1998', | ||||
| 'Altitude (km)', | ||||
| tmpplot, | ||||
| size = 'large', | ||||
| minColormap = 9, | ||||
| maxColormap = 12) | ||||
| except: | ||||
| traceback.print_exc() | ||||
| print 'FAILURE: plotting failed' | ||||
| sys.exit(-1) | ||||
| os.system('display %s' % (tmpplot)) | ||||
| os.remove(tmpplot) | ||||
| print 'SUCCESS - Madrigal build completed successfully' | ||||
