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()))