def test_split_series_using_lytaf():
'''test the downloading of the LYTAF file and subsequent queries'''
tmp_dir = tempfile.mkdtemp()
lyra.download_lytaf_database(lytaf_dir=tmp_dir)
assert os.path.exists(os.path.join(tmp_dir, 'annotation_ppt.db'))
# test split_series_using_lytaf
# construct a dummy signal for testing purposes
basetime = parse_time('2010-06-13 02:00')
seconds = 3600
dummy_time = [basetime + datetime.timedelta(0, s) for s in range(seconds)]
dummy_data = np.random.random(seconds)
lytaf_tmp = lyra.get_lytaf_events('2010-06-13 02:00',
'2010-06-13 06:00',
lytaf_path=tmp_dir,
combine_files=["ppt"])
split = lyra.split_series_using_lytaf(dummy_time, dummy_data, lytaf_tmp)
assert type(split) == list
assert len(split) == 4
assert split[0]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 0)
assert split[0]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 7, 2)
assert split[3]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 59, 41)
assert split[3]['subtimes'][-1] == datetime.datetime(
2010, 6, 13, 2, 59, 58)
# Test case when no LYTAF events found in time series.
split_no_lytaf = lyra.split_series_using_lytaf(dummy_time, dummy_data,
LYTAF_TEST)
assert type(split_no_lytaf) == list
assert type(split_no_lytaf[0]) == dict
assert not set(split_no_lytaf[0].keys()).symmetric_difference(
{'subtimes', 'subdata'})
assert split_no_lytaf[0]["subtimes"] == dummy_time
assert split_no_lytaf[0]["subdata"].all() == dummy_data.all()
def test_split_series_using_lytaf():
'''test the downloading of the LYTAF file and subsequent queries'''
tmp_dir = tempfile.mkdtemp()
lyra.download_lytaf_database(lytaf_dir=tmp_dir)
assert os.path.exists(os.path.join(tmp_dir, 'annotation_ppt.db'))
#test split_series_using_lytaf
#construct a dummy signal for testing purposes
basetime = parse_time('2010-06-13 02:00')
seconds = 3600
dummy_time = [basetime + datetime.timedelta(0, s) for s in range(seconds)]
dummy_data = np.random.random(seconds)
lytaf_tmp = lyra.get_lytaf_events('2010-06-13 02:00', '2010-06-13 06:00',
lytaf_path=tmp_dir,
combine_files=["ppt"])
split = lyra.split_series_using_lytaf(dummy_time, dummy_data, lytaf_tmp)
assert type(split) == list
assert len(split) == 4
assert split[0]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 0)
assert split[0]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 7, 2)
assert split[3]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 59, 41)
assert split[3]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 59, 58)
# Test case when no LYTAF events found in time series.
split_no_lytaf = lyra.split_series_using_lytaf(dummy_time,
dummy_data, LYTAF_TEST)
assert type(split_no_lytaf) == list
assert type(split_no_lytaf[0]) == dict
assert split_no_lytaf[0].keys() == ['subtimes', 'subdata']
assert split_no_lytaf[0]["subtimes"] == dummy_time
assert split_no_lytaf[0]["subdata"].all() == dummy_data.all()
def test_lytaf_utils():
'''test the downloading of the LYTAF file and subsequent queries'''
tmp_dir = tempfile.mkdtemp()
lyra.download_lytaf_database(lytaf_dir=tmp_dir)
assert os.path.exists(os.path.join(tmp_dir, 'annotation_ppt.db'))
#try doing a query on the temporary database
lar = lyra.get_lytaf_events(TimeRange('2010-06-13 02:00',
'2010-06-13 06:00'),
lytaf_dir=tmp_dir)
assert type(lar) == list
assert type(lar[0]) == dict
assert type(lar[0]['start_time']) == datetime.datetime
assert type(lar[0]['end_time']) == datetime.datetime
assert type(lar[0]['roi_description']) == str
assert type(lar[0]['event_type_description']) == str
assert lar[0]['start_time'] == parse_time('2010-06-13 02:07:04')
assert lar[0]['end_time'] == parse_time('2010-06-13 02:10:04')
assert lar[0]['event_type_description'] == 'LAR'
#test split_series_using_lytaf
#construct a dummy signal for testing purposes
basetime = parse_time('2010-06-13 02:00')
seconds = 3600
dummy_time = [basetime + datetime.timedelta(0, s) for s in range(seconds)]
dummy_data = np.random.random(seconds)
split = lyra.split_series_using_lytaf(dummy_time, dummy_data, lar)
assert type(split) == list
assert len(split) == 4
assert split[0]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 0)
assert split[0]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 7, 2)
assert split[3]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 59, 41)
assert split[3]['subtimes'][-1] == datetime.datetime(
2010, 6, 13, 2, 59, 58)
def test_lytaf_utils():
'''test the downloading of the LYTAF file and subsequent queries'''
tmp_dir=tempfile.mkdtemp()
lyra.download_lytaf_database(lytaf_dir=tmp_dir)
assert os.path.exists(os.path.join(tmp_dir,'annotation_ppt.db'))
#try doing a query on the temporary database
lar=lyra.get_lytaf_events(TimeRange('2010-06-13 02:00','2010-06-13 06:00'),lytaf_dir=tmp_dir)
assert type(lar) == list
assert type(lar[0]) == dict
assert type(lar[0]['start_time']) == datetime.datetime
assert type(lar[0]['end_time']) == datetime.datetime
assert type(lar[0]['roi_description']) == str
assert type(lar[0]['event_type_description']) == str
assert lar[0]['start_time'] == parse_time('2010-06-13 02:07:04')
assert lar[0]['end_time'] == parse_time('2010-06-13 02:10:04')
assert lar[0]['event_type_description'] == 'LAR'
#test split_series_using_lytaf
#construct a dummy signal for testing purposes
basetime=parse_time('2010-06-13 02:00')
seconds=3600
dummy_time = [basetime + datetime.timedelta(0, s) for s in range(seconds)]
dummy_data=np.random.random(seconds)
split=lyra.split_series_using_lytaf(dummy_time, dummy_data, lar)
assert type(split) == list
assert len(split) == 4
assert split[0]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 0)
assert split[0]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 7, 2)
assert split[3]['subtimes'][0] == datetime.datetime(2010, 6, 13, 2, 59, 41)
assert split[3]['subtimes'][-1] == datetime.datetime(2010, 6, 13, 2, 59, 58)
def test_lytaf_utils():
'''test the downloading of the LYTAF file and subsequent queries'''
tmp_dir=tempfile.mkdtemp()
lyra.download_lytaf_database(lytaf_dir=tmp_dir)
assert os.path.exists(os.path.join(tmp_dir,'annotation_ppt.db'))
#try doing a query on the temporary database
lar=lyra.get_lytaf_events(TimeRange('2010-06-13 02:00','2010-06-13 06:00'),lytaf_dir=tmp_dir)
assert type(lar) == list
assert type(lar[0]) == dict
assert type(lar[0]['start_time']) == datetime.datetime
assert type(lar[0]['end_time']) == datetime.datetime
assert type(lar[0]['roi_description']) == str
assert type(lar[0]['event_type_description']) == str
assert lar[0]['start_time'] == parse_time('2010-06-13 02:07:04')
assert lar[0]['end_time'] == parse_time('2010-06-13 02:10:04')
assert lar[0]['event_type_description'] == 'LAR'
