

@pytest.mark.parametrize('exname,colno', [(exname, colno) for exname in examples.keys() for colno in range(1, 3)])
def test_onecolumn(exname, colno):
    if ((exname == 't0') and (colno > 0)):
        pytest.skip(('Not enough columns in %s.' % (exname,)))
    if (exname.startswith('t1_sub') and (colno > 1)):
        pytest.skip(('Not enough columns in %s.' % (exname,)))
    with analyzed_bayesdb_generator(examples[exname](), 1, 1) as (bdb, generator_id):
        bqlfn.bql_column_value_probability(bdb, generator_id, None, colno, 4)
        bdb.sql_execute('select bql_column_value_probability(?, NULL, ?, 4)', (generator_id, colno)).fetchall()

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