

def test_prevalence_counter():
    counter = PrevalenceCounter()
    sql = 'SELECT * FROM foo WHERE bar GROUP BY baz;\n             select * from foo;\n             SELECT * FROM foo WHERE bar GROUP\n             BY baz'
    counter.update(names)
    keywords = ['SELECT', 'FROM', 'GROUP BY']
    expected = [3, 3, 2]
    kw_counts = [counter.keyword_count(x) for x in keywords]
    assert (kw_counts == expected)
    assert (counter.keyword_count('NOSUCHKEYWORD') == 0)
    names = ['foo', 'bar', 'baz']
    name_counts = [counter.name_count(x) for x in names]
    assert (name_counts == [3, 2, 2])

dataset/ETHPy150Open dbcli/pgcli/tests/test_prioritization.py test_prevalence_counter/VarMisuse@23/81 `sql`->`names`