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

rate_eff_cov_table.py

View page source

rate_eff_cov_table: Example and Test

def rate_eff_cov_table() :
    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 rate table
    col_name = [
        'covariate_id', 'node_id', 'split_value', 'weight_id'
    ]
    col_type = [
        'integer',      'integer', 'real',        'integer'
    ]
    # sex is our splitting covariate and -0.5 corresponds to female.
    row_list = [
        [1,             0,          -0.5,         1],
        [1,             1,          -0.5,         0],
        [1,             2,          -0.5,         1],
        [1,             3,          -0.5,         0],
    ]
    tbl_name = 'rate_eff_cov'
    dismod_at.create_table(connection, tbl_name, col_name, col_type, row_list)
    # ----------------------------------------------------------------------
    # include primary key in test
    check_name = [ tbl_name + '_id' ] + col_name
    check_list = list()
    for i in range( len(row_list) ) :
        check_list.append( [i] + row_list[i] )
    #
    row_list = dismod_at.get_row_list(connection, tbl_name, check_name)
    assert row_list == check_list
    # ----------------------------------------------------------------------
    connection.close()
    print('rate_eff_cov_table: OK')