Adding StationXML test case.

Author: krischer

stationxml_test/README.md

@@ -0,0 +1,19 @@
+# StationXML Test Case
+
+This folder contains the test case for the evalresp integration in ObsPy.
+
+The following is a short description of the files.
+
+* **download_all_stations.py**: Downloads the StationXML files at the response
+  level from IRIS. It requires an `all_stations.xml` file, a station level
+  StationXML file which has to be downloaded separately.
+* **convert_to_SEED.sh**: Bash script converting all StationXML files to SEED
+  using the Java tool by IRIS.
+* **evresp_process.py**: Running evalresp in a separate process to test whether
+  it segfaults or not. Otherwise it would crash the current Python process.
+
+* **test_response_large_scale.py**: The actual test case. It loops over every
+  StationXML file in the *StationXML* subfolder and calculates the response for
+  each channel using the ObsPy to evalresp bridge. These responses are compared
+  to responses calculate by converting the SEED files in the *SEED* subfolder
+  to RESP files and directly using evalresp with them.

stationxml_test/convert_to_SEED.sh

@@ -0,0 +1,25 @@
+#!/usr/bin/env bash
+# -*- coding: utf-8 -*-
+#
+# Converts all StationXML files to SEED files.
+
+folder="SEED/"
+extension=".seed"
+converter="~/Downloads/stationxml-converter-1.0.1.jar"
+
+for i in StationXML/*.xml
+    do
+        name=`echo $i | cut -d'/' -f2`
+
+        filename=$(basename "$name")
+        filename="${filename%.*}"
+
+        name=$folder$filename$extension
+
+        if [ ! -f $name ]
+        then
+            echo "Converting " $i
+            java -jar $converter -s $i > $name
+        fi
+
+    done

stationxml_test/download_all_stations.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Downloads the response level StationXML file for all stations in the file
+'all_stations.xml' which contains a station level StationXML file.
+
+:copyright:
+    Lion Krischer ([email protected]), 2014
+:license:
+    GNU Lesser General Public License, Version 3
+    (http://www.gnu.org/copyleft/lesser.html)
+"""
+import colorama
+from obspy.station import read_inventory
+from obspy.fdsn import Client
+import os
+
+output_dir = "StationXML"
+
+c = Client()
+
+inv = read_inventory("./all_stations.xml")
+
+
+def print_error(msg):
+    print colorama.Fore.RED + msg + colorama.Fore.RESET
+
+
+def print_ok(msg):
+    print colorama.Fore.GREEN + msg + colorama.Fore.RESET
+
+
+for network in inv.networks:
+    for station in network.stations:
+        output_filename = os.path.join(output_dir, "%s.%s.xml" %
+                                       (network.code, station.code))
+        if os.path.exists(output_filename):
+            continue
+
+        try:
+            out = c.get_stations(network=network.code, station=station.code,
+                                 level="response")
+        except:
+            print_error("Failed to download %s.%s." % (network.code,
+                                                       station.code))
+            continue
+        with open(output_filename, "w") as fh:
+            fh.write(out)
+        print_ok("Downloaded %s.%s." % (network.code, station.code))

stationxml_test/evresp_process.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Separate process to test for evalresp segfaults which otherwise interrupt the
+current Python process.
+
+:copyright:
+    Lion Krischer ([email protected]), 2014
+:license:
+    GNU Lesser General Public License, Version 3
+    (http://www.gnu.org/copyleft/lesser.html)
+"""
+from multiprocessing import Process
+from obspy.signal.invsim import evalresp
+
+from obspy.core.util.misc import CatchOutput
+
+import sys
+
+
+def test_evalresp_segfault(t_samp, filename, date, stat_id, chan_id, net_id,
+                           loc_id, units):
+        out = None
+        try:
+            with CatchOutput() as out:
+                evalresp(
+                    t_samp, 5, filename, date=date, station=stat_id,
+                    channel=chan_id, network=net_id, locid=loc_id,
+                    units=units, freq=True)
+        except:
+            if out and out.stderr and "are not supported" in out.stderr:
+                sys.exit(99)
+            else:
+                sys.exit(1)
+
+
+def test_for_segfault(t_samp, filename, date, stat_id, chan_id, net_id, loc_id,
+                      units):
+
+    p = Process(target=test_evalresp_segfault,
+                args=(t_samp, filename, date, stat_id, chan_id, net_id,
+                      loc_id, units))
+    p.start()
+    p.join()
+
+    p.terminate()
+    p.terminate()
+    p.terminate()
+
+    exitcode = p.exitcode
+    filename.seek(0, 0)
+
+    if exitcode and exitcode not in [1, 99]:
+        return "segfault"
+    elif exitcode == 1:
+        return "Failed to calculate response"
+    elif exitcode == 99:
+        return "uniterror"
+    else:
+        return None