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'