def format_data(self, query, timeseries, start_response, environ):
"""Format requested timeseries.
Parameters
----------
query : dictionary of parsed query parameters
timeseries : obspy.core.Stream
timeseries object with data to be written
Returns
-------
unicode
IAGA2002 formatted string.
"""
url = environ['HTTP_HOST'] + environ['PATH_INFO'] + \
environ['QUERY_STRING']
if query.output_format == 'json':
timeseries_string = IMFJSONWriter.format(timeseries,
query.elements, url)
else:
timeseries_string = IAGA2002Writer.format(timeseries,
query.elements)
start_response('200 OK',
[
("Content-Type", "text/plain")
])
return timeseries_string
def test_times():
"""imfjson.IMFJSONWriter_test.test_times()
Call the _format_times method with the test
data and channels.
Verify, the times are the correct value and string format.
"""
writer = IMFJSONWriter()
times = writer._format_times(EXAMPLE_DATA, EXAMPLE_CHANNELS)
# load times to test against
test_day, test_time = np.genfromtxt(EXAMPLE_FILE, skip_header=25,
usecols=(0, 1), unpack=True, dtype=str)
test_date_times = []
for idx in range(test_day.shape[0]):
test_date_times += [test_day[idx] + "T" + test_time[idx] + "Z"]
assert_equals(times, test_date_times)
def test_times():
"""imfjson.IMFJSONWriter_test.test_times()
Call the _format_times method with the test
data and channels.
Verify, the times are the correct value and format.
"""
writer = IMFJSONWriter()
times = writer._format_times(EXAMPLE_DATA, EXAMPLE_CHANNELS)
test_day, test_time = np.loadtxt(EXAMPLE_FILE,
skiprows=25,
usecols=(0, 1),
unpack=True,
dtype=str)
test_date_times = []
for idx in range(test_day.shape[0]):
test_date_times += [test_day[idx] + "T" + test_time[idx] + "Z"]
assert_equals(times, test_date_times)
def test_values():
"""imfjson.IMFJSONWriter_test.test_values()
Call the _format_data method with the test
data, channels, and stats.
Verify, the values and associated metadata
are the correct value and format.
"""
writer = IMFJSONWriter()
values = writer._format_data(EXAMPLE_DATA, EXAMPLE_CHANNELS, EXAMPLE_STATS)
test_val_keys = ["id", "metadata", "values"]
for val in values:
for key, test in zip(val, test_val_keys):
assert_equals(key, test)
assert_equals(values[0]['id'], "H")
assert_equals(values[1]['id'], "D")
assert_equals(values[2]['id'], "Z")
assert_equals(values[3]['id'], "F")
# Test values-metadata (need to add flags)
metadata = values[0]['metadata']
test_metadata_keys = [
"element", "network", "station", "channel", "location"
]
for key, test in zip(metadata, test_metadata_keys):
assert_equals(key, test)
assert_equals(metadata['element'], "H")
assert_equals(metadata['network'], "NT")
assert_equals(metadata['station'], "BOU")
# channels do not match H v MVH
# assert_equals(metadata['channel'], "MVH")
assert_equals(metadata['location'], "R0")
# Test values-values
# Round to match iaga format
vals_H = np.around(values[0]['values'], 2)
vals_D = np.around(values[1]['values'], 2)
test_val_H, test_val_D = np.loadtxt(EXAMPLE_FILE,
skiprows=25,
usecols=(3, 4),
unpack=True,
dtype=float)
# tolist required to prevent ValueError in comparison
assert_equals(vals_H.tolist(), test_val_H.tolist())
assert_equals(vals_D.tolist(), test_val_D.tolist())
def test_values():
"""imfjson.IMFJSONWriter_test.test_values()
Call the _format_data method with the test
data, channels, and stats.
Verify, the values and associated metadata
are the correct value and format.
"""
writer = IMFJSONWriter()
values = writer._format_data(EXAMPLE_DATA, EXAMPLE_CHANNELS,
EXAMPLE_STATS)
test_val_keys = ["id", "metadata", "values"]
for val in values:
for key, test in zip(val, test_val_keys):
assert_equals(key, test)
assert_equals(values[0]['id'], "H")
assert_equals(values[1]['id'], "D")
assert_equals(values[2]['id'], "Z")
assert_equals(values[3]['id'], "F")
# Test values-metadata (need to add flags)
metadata = values[0]['metadata']
test_metadata_keys = ["element", "network", "station",
"channel", "location"]
for key, test in zip(metadata, test_metadata_keys):
assert_equals(key, test)
assert_equals(metadata['element'], "H")
assert_equals(metadata['network'], "NT")
assert_equals(metadata['station'], "BOU")
# channels do not match H v MVH
# assert_equals(metadata['channel'], "MVH")
assert_equals(metadata['location'], "R0")
# Test values-values
# Round to match iaga format
vals_H = np.around(values[0]['values'], 2)
vals_D = np.around(values[1]['values'], 2)
test_val_H, test_val_D = np.loadtxt(EXAMPLE_FILE, skiprows=25,
usecols=(3, 4), unpack=True, dtype=float)
# tolist required to prevent ValueError in comparison
assert_equals(vals_H.tolist(), test_val_H.tolist())
assert_equals(vals_D.tolist(), test_val_D.tolist())
def test_metadata():
"""imfjson.IMFJSONWriter_test.test_metadata()
Call the _format_metadata method with the test
stats and channels.
Verify, the set metadata.
"""
writer = IMFJSONWriter()
metadata = writer._format_metadata(EXAMPLE_STATS, EXAMPLE_CHANNELS)
assert_equals(metadata['status'], 200)
# Test intermagnet parameters
intermag = metadata['intermagnet']
assert_equals(intermag['reported_orientation'], "HDZF")
assert_equals(intermag['sensor_orientation'], "HDZF")
assert_equals(intermag['data_type'], "variation")
assert_equals(intermag['sampling_period'], 60)
assert_equals(intermag['digital_sampling_rate'], 0.01)
# Test intermagnet-imo parameters
imo = metadata['intermagnet']['imo']
assert_equals(imo['iaga_code'], "BOU")
assert_equals(imo['name'], "Boulder")
assert_equals(imo['coordinates'], [254.764, 40.137, 1682])
def test_metadata():
"""imfjson.IMFJSONWriter_test.test_metadata()
Call the _format_metadata method with the test
stats and channels.
Verify, the set metadata.
"""
writer = IMFJSONWriter()
metadata = writer._format_metadata(EXAMPLE_STATS, EXAMPLE_CHANNELS)
assert_equals(metadata['status'], 200)
# Test intermagnet parameters
intermag = metadata['intermagnet']
assert_equals(intermag['reported_orientation'], "HDZF")
assert_equals(intermag['sensor_orientation'], "HDZF")
assert_equals(intermag['data_type'], "variation")
assert_equals(intermag['sampling_period'], 60)
assert_equals(intermag['digital_sampling_rate'], 0.01)
# Test intermagnet-imo parameters
imo = metadata['intermagnet']['imo']
assert_equals(imo['iaga_code'], "BOU")
assert_equals(imo['name'], "Boulder")
assert_equals(imo['coordinates'], [254.764, 40.137, 1682])
def test_metadata():
"""imfjson.IMFJSONWriter_test.test_metadata()
Call the _format_metadata method with the test
stats and channels.
Verify, the set metadata.
"""
writer = IMFJSONWriter()
metadata = writer._format_metadata(EXAMPLE_STATS, EXAMPLE_CHANNELS)
assert_equal(metadata["status"], 200)
# Test intermagnet parameters
intermag = metadata["intermagnet"]
assert_equal(intermag["reported_orientation"], "HDZF")
assert_equal(intermag["sensor_orientation"], "HDZF")
assert_equal(intermag["data_type"], "variation")
assert_equal(intermag["sampling_period"], 60)
assert_equal(intermag["digital_sampling_rate"], 0.01)
# Test intermagnet-imo parameters
imo = metadata["intermagnet"]["imo"]
assert_equal(imo["iaga_code"], "BOU")
assert_equal(imo["name"], "Boulder")
assert_equal(imo["coordinates"], [254.764, 40.137, 1682])
def format_data(self, query, timeseries, start_response, environ):
"""Format requested timeseries.
Parameters
----------
query : dictionary of parsed query parameters
timeseries : obspy.core.Stream
timeseries object with data to be written
Returns
-------
unicode
IAGA2002 formatted string.
"""
url = environ["HTTP_HOST"] + environ["PATH_INFO"] + environ[
"QUERY_STRING"]
if query.output_format == "json":
timeseries_string = IMFJSONWriter.format(timeseries,
query.elements, url)
else:
timeseries_string = IAGA2002Writer.format(timeseries,
query.elements)
start_response("200 OK", [("Content-Type", "text/plain")])
return timeseries_string