def test_prepare_data_duplicate_channels(self):
data = self.data.copy()
data += data[0].copy()
with self.assertRaises(LagCalcError):
_prepare_data(family=self.party[0],
detect_data=data,
shift_len=0.2)
def test_duplicate_channel_error(self):
detect_data = self.detection + self.detection
with self.assertRaises(LagCalcError):
_prepare_data(detect_data=detect_data,
detections=[self.detections[0]],
template=('test_template', self.template),
delays=self.delays,
shift_len=0.5,
plot=False)
def test_no_matching_template(self):
detect_streams = _prepare_data(detect_data=self.detection,
detections=[self.detections[0]],
template=('fake_template',
self.template),
delays=self.delays,
shift_len=0.5,
plot=False)
self.assertEqual(len(detect_streams), 0)
def test_prepare_data(self):
shift_len = 0.2
for family in self.party:
detect_stream_dict = _prepare_data(family=family,
detect_data=self.data,
shift_len=shift_len)
self._prepare_data_checks(detect_stream_dict=detect_stream_dict,
family=family,
shift_len=shift_len)
def test_prepare_data_too_short(self):
data = self.data.copy()
data.trim(self.party[0][0].detect_time,
self.party[0][0].detect_time + 3)
shift_len = 0.2
detect_stream_dict = _prepare_data(family=self.party[0],
detect_data=data,
shift_len=shift_len)
self.assertEqual(len(detect_stream_dict), 1)
def test_merged_data(self):
detect_data = self.detection.copy()
detect_data.cutout(starttime=detect_data[0].stats.starttime + 10,
endtime=detect_data[0].stats.starttime +
12).merge()
detect_streams = _prepare_data(detect_data=detect_data,
detections=[self.detections[0]],
template=('fake_template',
self.template),
delays=self.delays,
shift_len=0.5,
plot=False)
self.assertEqual(len(detect_streams), 0)
def test_prepare_data_masked(self):
data = self.data.copy()
data.cutout(self.party[0][0].detect_time,
self.party[0][0].detect_time + 3)
data.merge()
shift_len = 0.2
detect_stream_dict = _prepare_data(family=self.party[0],
detect_data=data,
shift_len=shift_len)
short_key = self.party[0][0].id
for key, value in detect_stream_dict.items():
detection = [d for d in self.party[0] if d.id == key][0]
if key == short_key:
self.assertNotEqual(len(value), detection.no_chans)
else:
self.assertEqual(len(value), detection.no_chans)
