def test_get_data_no_wildcard(self, file_mock, read_hdf5_header_mock):
read_hdf5_header_mock.return_value = self.rftr.stats
net = 'AB'
stat = 'CD'
tag = 'rand'
phase = 'P'
pol = 'v'
evt_time = UTCDateTime(0)
exp_path = rfh5.hierarchy.format(tag=tag,
network=net,
station=stat,
phase=phase,
pol=pol,
evt_time=evt_time.format_fissures())
d = {exp_path: self.rftr.data}
file_mock.side_effect = d.__getitem__
self.assertTrue(np.all(self.dbh[exp_path] == d[exp_path]))
outdata = self.dbh.get_data(net,
stat,
phase,
evt_time,
tag=tag,
pol=pol)
self.assertEqual(outdata[0], self.rftr)
file_mock.assert_called_with(exp_path)
def test_get_data_wildcard(self, file_mock, all_tr_recursive_mock):
all_tr_recursive_mock.return_value = None
net = 'AB'
stat = '*'
tag = 'rand'
phase = 'P'
pol = 'v'
evt_time = UTCDateTime(0)
exp_path = rfh5.hierarchy.format(tag=tag,
network=net,
station=stat,
phase=phase,
pol=pol,
evt_time=evt_time.format_fissures())
d = {exp_path: self.rftr.data, '/rand/AB/': self.rftr.data}
file_mock.side_effect = d.__getitem__
_ = self.dbh.get_data(net, stat, phase, evt_time, tag=tag, pol=pol)
file_mock.assert_called_with('/rand/AB/')
all_tr_recursive_mock.assert_called_with(
d['/rand/AB/'], RFStream(),
'/rand/AB*/%s/%s/%s' % (phase, pol, evt_time.format_fissures()))
