Exemple #1
0
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'
Exemple #2
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'