def test_commit(): '''Commit request and retrieve data from h5 files''' my_dr = dr.DataReader(data_path, 'out') my_dr.get_files_info() test_data = my_dr._get_data("2015-05-30 11:26:00", "2015-08-30") tags = dr.get_tags(test_data.columns, elecs=['2101A'], cells=[100]) tags_x = [el for el in tags if ".U." not in el] tags_y = [el for el in tags if ".U." in el] print(tags_x, tags_y) size = 50 shape_x = (-1, size, len(tags_x) + 1) shape_y = (-1, len(tags_y)) shape_endog = (-1, size, len(tags_y)) print(shape_x, shape_y) data_s = dict() data_s['bDate'] = "2015-05-30 11:26:00" data_s['eDate'] = "2015-08-30" exog = {'tags': tags_x, 'shape': shape_x, 'index': True, 'patched': True, 'norm': True, 'size':size} endog = {'tags': tags_y, 'shape': shape_endog, 'index': False, 'patched': True, 'norm': True, 'size':size} output = {'tags': tags_y, 'shape': shape_y, 'index': False, 'patched': False, 'norm': False, 'size':size} data_s['tags_struc'] = {'exog': exog, 'endog': endog, 'output': output} data = my_dr.commit(data_s) assert len(data) > 0, 'Data len is 0'
def test_get_tags(): '''Get tags from h5 files''' my_dr = dr.DataReader(data_path, 'out') my_dr.get_files_info() test_data = my_dr._get_data("2015-05-30 11:26:00", "2015-08-30") tags = dr.get_tags(test_data.columns, elecs=['2101A'], cells=[100]) assert len(tags) > 0, 'Tags len is 0'