def test_parse_dict_null_qc_mask(self):
output = io.StringIO()
sys.stdout = output
data_object = 'qc-mask'
parse_dict(self.null, data_object)
sys.stdout = sys.__stdout__
self.assertEqual(output.getvalue(), "Error: Service Returned Null Qc-Mask Data\n")
def test_parse_dict_null_channel(self):
output = io.StringIO()
sys.stdout = output
data_object = 'channel'
parse_dict(self.null, data_object)
sys.stdout = sys.__stdout__
self.assertEqual(output.getvalue(), "Error: Service Returned Null Channel Data\n")
def test_parse_dict_null_soh(self):
output = io.StringIO()
sys.stdout = output
data_object = 'acquired-channel-soh'
parse_dict(self.null, data_object)
sys.stdout = sys.__stdout__
self.assertEqual(output.getvalue(), "Error: Service Returned Null Acquired-Channel-Soh Data\n")
def test_parse_dict_wrong_object(self):
output = io.StringIO()
sys.stdout = output
data_object = 'wrong-object'
parse_dict(self.null, data_object)
sys.stdout = sys.__stdout__
self.assertEqual(output.getvalue(), "Error: parse_dict(): data_object not found\n")
def setUp(self):
self.soh_json = read_json(os.path.dirname(__file__) + '/acquired_channel_soh_mock.json')
self.channel = read_json(os.path.dirname(__file__) + '/channel_mock.json')
self.channel_segment = read_json(os.path.dirname(__file__) + '/channel_segment_mock.json')
self.qc_mask_json = read_json(os.path.dirname(__file__) + '/qc_mask_mock.json')
self.chan_seg = parse_dict(self.channel_segment, 'channel-segment')
self.chan = parse_dict(self.channel, 'channel')
self.soh = parse_dict(self.soh_json, 'acquired-channel-soh')
self.qcmask = parse_dict(self.qc_mask_json, 'qc-mask')
def setUp(self):
self.date = '1969-12-31T17:00:00Z'
self.channel = read_json(
os.path.dirname(__file__) + '/channel_mock.json')
self.channel_segment = read_json(
os.path.dirname(__file__) + '/channel_segment_mock.json')
self.chan_seg = parse_dict(self.channel_segment, 'channel-segment')
self.chan = parse_dict(self.channel, 'channel')
self.dfile_base = 'waveform'
self.dir = os.path.dirname(__file__)
self.wfdisc = 'test.wfdisc'
self.wfdisc_wav = 'test_one_wave.wfdisc'
self.chan_seg_one_wv = parse_dict(self.channel_segment,
'channel-segment')
self.chan_seg_one_wv.waveform = [self.chan_seg_one_wv.waveform[0]]
def test_parse_dict_soh(self):
data_object = 'acquired-channel-soh'
soh = parse_dict(self.soh_json, data_object)
self.assertEqual(soh[0].processing_channel_id, self.soh_json[0]['processingChannelId'])
self.assertEqual(soh[0].acquired_channel_soh_type, self.soh_json[0]['type'])
self.assertEqual(soh[0].start_time, self.soh_json[0]['startTime'])
self.assertEqual(soh[0].end_time, self.soh_json[0]['endTime'])
self.assertEqual(soh[0].status_type, self.soh_json[0]['status'])
def test_parse_dict_channel_segment(self):
data_object = 'channel-segment'
chan_seg = parse_dict(self.channel_segment, data_object)
self.assertEqual(chan_seg.name, self.channel_segment['name'])
self.assertEqual(chan_seg.processing_channel_id, self.channel_segment['processingChannelId'])
self.assertEqual(chan_seg.id, self.channel_segment['id'])
self.assertEqual(chan_seg.channel_segment_type, self.channel_segment['segmentType'])
self.assertEqual(chan_seg.start_time, self.channel_segment['startTime'])
self.assertEqual(chan_seg.end_time, self.channel_segment['endTime'])
self.assertEqual(chan_seg.waveform[0].end_time, self.channel_segment['waveforms'][0]['endTime'])
self.assertEqual(chan_seg.waveform[0].start_time, self.channel_segment['waveforms'][0]['startTime'])
self.assertEqual(chan_seg.waveform[0].sample_count, self.channel_segment['waveforms'][0]['sampleCount'])
self.assertEqual(chan_seg.waveform[0].sample_rate, self.channel_segment['waveforms'][0]['sampleRate'])
self.assertEqual(chan_seg.waveform[0].values, self.channel_segment['waveforms'][0]['values'])
def test_parse_dict_channel(self):
data_object = 'channel'
chan = parse_dict(self.channel, data_object)
self.assertEqual(chan.name, self.channel['name'])
self.assertEqual(chan.channel_type, self.channel['type'])
self.assertEqual(chan.sites.name, self.channel['sites']['name'])
self.assertEqual(chan.sites.channels, self.channel['sites']['channels'])
self.assertEqual(chan.sites.station.name, self.channel['sites']['station']['name'])
self.assertEqual(chan.sites.station.elevation, self.channel['sites']['station']['elevation'])
self.assertEqual(chan.sites.station.longitude, self.channel['sites']['station']['longitude'])
self.assertEqual(chan.sites.station.latitude, self.channel['sites']['station']['latitude'])
self.assertEqual(chan.sites.station.station_type, self.channel['sites']['station']['stationType'])
self.assertEqual(chan.instruments.instrument_model, self.channel['instruments']['instrumentModel'])
self.assertEqual(chan.calibration.calibration_period, self.channel['calibration']['period'])
self.assertEqual(chan.calibration.calibration_factor, self.channel['calibration']['factor'])
if channelid is None:
print('Parameter channel_id is required for metadata')
raise TypeError
starttime = format_time(starttime, startdate)
endtime = format_time(endtime, enddate)
params = set_params(channel_id=channelid,
start_time=starttime,
end_time=endtime,
with_waveforms=False)
channel_segment = http_request(None, 'channel-segment', params, 'json')
# channel = http_request('channel', params, 'json')
# mocked
channel = http_mock_request('channel')
# channel_segment = http_mock_request('channel-segment')
chan_seg = parse_dict(channel_segment, 'channel-segment')
chan = parse_dict(channel, 'channel')
if table is True:
table_waveform(chan_seg, chan)
elif data_type == 'waveforms':
if channelid is None:
print('Parameter channel_id is required for waveforms')
raise TypeError
starttime = format_time(starttime, startdate)
endtime = format_time(endtime, enddate)
params = set_params(channel_id=channelid,
start_time=starttime,
end_time=endtime,
with_waveforms=True)
def test_parse_dict_qc_mask(self):
data_object = 'qc-mask'
qc = parse_dict(self.qc_mask_json, data_object)
self.assertEqual(qc[0].id, self.qc_mask_json[0]['id'])
self.assertEqual(qc[0].processing_channel_id, self.qc_mask_json[0]['processingChannelId'])