def test_get_timeseries(self): # test simple get after add a = TimeSeries([1, 2, 3, 4, 5], name='test name', epoch=0, sample_rate=1) data.add_timeseries(a) b = data.get_timeseries('test name', [(0, 5)]) self.assertEqual(a, b) # test more complicated add with a cache a = data.get_timeseries('H1:LOSC-STRAIN', LOSC_SEGMENTS, cache=self.FRAMES['H1:LOSC-STRAIN']) b = data.get_timeseries('H1:LOSC-STRAIN', LOSC_SEGMENTS) self.assertEqual(a, b)
def test_get_timeseries(self): # empty globalv.DATA globalv.DATA = type(globalv.DATA)() # test simple get after add a = TimeSeries([1, 2, 3, 4, 5], name='test name', epoch=0, sample_rate=1) data.add_timeseries(a) b, = data.get_timeseries('test name', [(0, 5)], nproc=1) nptest.assert_array_equal(a.value, b.value) assert a.sample_rate.value == b.sample_rate.value # test more complicated add with a cache a, = data.get_timeseries('H1:LOSC-STRAIN', LOSC_SEGMENTS, cache=self.FRAMES['H1:LOSC-STRAIN'], nproc=1) b, = data.get_timeseries('H1:LOSC-STRAIN', LOSC_SEGMENTS, nproc=1) nptest.assert_array_equal(a.value, b.value)
def test_read_archive(self): fname = self.test_write_archive(delete=False) try: archive.read_data_archive(fname) finally: os.remove(fname) ts = data.get_timeseries('X1:TEST-CHANNEL', [(100, 110)], query=False).join() nptest.assert_array_equal(ts.value, TEST_DATA.value) for attr in ['epoch', 'unit', 'sample_rate', 'channel', 'name']: self.assertEqual(getattr(ts, attr), getattr(TEST_DATA, attr))
def test_read_archive(): fname = test_write_archive(delete=False) empty_globalv() try: archive.read_data_archive(fname) finally: os.remove(fname) # check timeseries ts = data.get_timeseries('X1:TEST-CHANNEL', [(100, 110)], query=False).join() nptest.assert_array_equal(ts.value, TEST_DATA.value) for attr in ['epoch', 'unit', 'sample_rate', 'channel', 'name']: assert getattr(ts, attr) == getattr(TEST_DATA, attr) # check trend series ts = data.get_timeseries('X1:TEST-TREND.mean,m-trend', [(0, 300)], query=False).join() assert ts.channel.type == 'm-trend' assert ts.span == (0, 300) # check triggers t = triggers.get_triggers('X1:TEST-TABLE', 'testing', [(0, 100)]) assert len(t) == 100