6eddfdb1d05359bb55b07060553d8e59310f8aa8,tests/test_bql.py,,test_using_models,#,1380

Before Change

        bql2sql("estimate pairwise mutual information from t1 using model 1")
    with pytest.raises(NotImplementedError):
        bql2sql("estimate pairwise row similarity from t1 using model 1")
    with pytest.raises(NotImplementedError):
        bql2sql("infer x, y from t1 using model 1")
    with pytest.raises(NotImplementedError):
        bql2sql("infer explicit x, y from t1 using model 1")

After Change

    def setup(bdb):
        bdb.execute("initialize 1 model for t1_cc")
        bdb.execute("analyze t1_cc for 1 iteration wait")
    assert bql2sql("simulate age, weight from t1_cc using model 0"
            " limit 1", setup=setup) == \
        "SELECT * FROM "bayesdb_temp_0";"
    assert bql2sql("estimate predictive probability of weight from t1_cc"
            " using model 42") == \
        "SELECT bql_row_column_predictive_probability(1, 42, _rowid_, 3)" \
            " FROM "t1";"

Instances