\(\newcommand{\B}[1]{ {\bf #1} }\) \(\newcommand{\R}[1]{ {\rm #1} }\) \(\newcommand{\W}[1]{ \; #1 \; }\)

get_row_list.py

View page source

get_row_list: Example and Test

def get_row_list() :
    import dismod_at
    import copy
    #
    file_name      = 'example.db'
    connection     = dismod_at.create_connection(
        file_name, new = True, readonly = False
    )
    cursor         = connection.cursor()
    #
    # create the covariate table
    col_name = [ 'covariate_name',  'reference' ]
    col_type = [ 'text',             'real'     ]
    row_list = [
                  [ 'sex',              0.0        ],
                  [ 'income',           2000.0     ]
    ]
    tbl_name = 'covariate'
    dismod_at.create_table(connection, tbl_name, col_name, col_type, row_list)
    n_row = len(row_list)
    #
    # reverse the order of the columns
    col_name = [ 'reference',   'covariate_name' ]
    n_col    = len(col_name)
    row_list = dismod_at.get_row_list(connection, tbl_name, col_name)
    #
    assert len(row_list) == n_row
    for i in range(n_row) :
        assert len(row_list[i] ) == n_col
        assert isinstance( row_list[i][0], float)
        assert isinstance( row_list[i][1], str)
    assert row_list[0][0] == 0.0
    assert row_list[0][1] == 'sex'
    assert row_list[1][0] == 2000.0
    assert row_list[1][1] == 'income'
    #
    connection.close()
    print('get_row_list: OK')