

@pytest.mark.parametrize('exname,colno0,colno1', [(exname, colno0, colno1) for exname in examples.keys() for colno0 in range(1, 3) for colno1 in range(1, 3)])
def test_twocolumn(exname, colno0, colno1):
    if (exname == 't0'):
        pytest.skip('Not enough columns in t0.')
    if (exname.startswith('t1_sub') and ((colno0 > 1) or (colno1 > 1))):
        pytest.skip(('Not enough columns in %s.' % (exname,)))
    with analyzed_bayesdb_generator(examples[exname](), 1, 1) as (bdb, generator_id):
        bqlfn.bql_column_correlation(bdb, generator_id, colno0, colno1)
        bdb.sql_execute('select bql_column_correlation(?, ?, ?)', (generator_id, colno0, colno1)).fetchall()
        bqlfn.bql_column_dependence_probability(bdb, generator_id, None, colno0, colno1)
        bdb.sql_execute('select bql_column_dependence_probability(?, NULL, ?, ?)', (generator_id, colno0, colno1)).fetchall()
        bqlfn.bql_column_mutual_information(bdb, generator_id, None, colno0, colno1)
        bqlfn.bql_column_mutual_information(bdb, generator_id, None, colno0, colno1, numsamples=None)
        bqlfn.bql_column_mutual_information(bdb, generator_id, None, colno0, colno1, numsamples=1)
        bdb.sql_execute('select bql_column_mutual_information(?, NULL, ?, ?, NULL)', (generator_id, colno0, colno1)).fetchall()
        bdb.sql_execute('select bql_column_mutual_information(?, NULL, ?, ?, 1)', (generator_id, colno0, colno1)).fetchall()
        bdb.sql_execute('select bql_column_mutual_information(?, NULL, ?, ?, 100)', (generator_id, colno0, colno1)).fetchall()

dataset/ETHPy150Open probcomp/bayeslite/tests/test_core.py test_twocolumn/original