Source code for seismic.xcorqc.validate_xcorr_setup

#!/usr/bin/env python
#
# Test that the user's Raijin setup is capable of importing and using require libraries
# for x-correlation in mpi runtime environment.
# Requires Python 2.7.13 and the following module loads before running:
#     module load openmpi/2.1.1
#     module load hdf5/1.10.2p
#     module load module load mpi4py/3.0.0-py2

import sys

def test_setup():
    try:
        print("Testing mpi4py import")
        from mpi4py import MPI
        print("SUCCESS!")
    except:
        print("Failed to import mpi4py")
        sys.exit(1)

    try:
        print("Testing h5py import")
        import h5py
        print("SUCCESS!")
    except:
        sys.exit(1)

    try:
        print("Testing obspy import")
        import obspy
        print("SUCCESS!")
    except:
        sys.exit(1)

    try:
        print("Testing pyasdf import")
        import pyasdf
        print("SUCCESS!")
    except:
        sys.exit(1)

    try:
        print("Testing netCDF4 import")
        from netCDF4 import Dataset
        print("SUCCESS!")
    except:
        sys.exit(1)

    try:
        print("Opening ASDF file for reading")
        afile = pyasdf.ASDFDataSet('../../tests/seismic/xcorqc/data/test_data_CMSA.h5', mode='r')
        print("SUCCESS!")
    except:
        print("Failed opening ASDF file")
        sys.exit(1)

    print(afile)

    try:
        print("Reading trace from ASDF file")
        stream, inv = afile.get_data_for_tag('AU.CMSA', 'raw_recording')
        print(inv)
        print(stream[0])
        print("SUCCESS!")
    except:
        print("Failed to read data from ASDF file")
        sys.exit(1)


if __name__ == '__main__':
    test_setup()