summaryrefslogtreecommitdiffstats
path: root/yql/essentials/sql/v1/ide/completion/name/service/ranking/frequency_ut.cpp
blob: b865e0799cc4ca617fef337d014aac868f6fb84e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#include "frequency.h"

#include <library/cpp/testing/unittest/registar.h>

using namespace NSQLComplete;

Y_UNIT_TEST_SUITE(FrequencyTests) {

Y_UNIT_TEST(FrequencyDataJson) {
    TFrequencyData actual = Pruned(ParseJsonFrequencyData(R"([
            {"parent":"FUNC","rule":"ABC","sum":1},
            {"parent":"TYPE","rule":"BIGINT","sum":7101},
            {"parent":"KEYWORD","rule":"UNION","sum":65064443},
            {"parent":"MODULE_FUNC","rule":"Compress::BZip2","sum":2},
            {"parent":"MODULE","rule":"re2","sum":3094},
            {"parent":"READ_HINT","rule":"COLUMNS","sum":826110},
            {"parent":"INSERT_HINT","rule":"COLUMN_GROUPS","sum":225},
            {"parent":"TRule_action_or_subquery_args","rule":"TRule_action_or_subquery_args.Block2","sum":4874480}
        ])"));

    TFrequencyData expected = {
        .Keywords = {
            {"union", 65064443},
        },
        .Types = {
            {"bigint", 7101},
        },
        .Functions = {
            {"abc", 1},
            {"compress::bzip2", 2},
        },
        .Hints = {
            {"columns", 826110},
            {"columngroups", 225},
        },
    };

    UNIT_ASSERT_VALUES_EQUAL(actual.Keywords, expected.Keywords);
    UNIT_ASSERT_VALUES_EQUAL(actual.Types, expected.Types);
    UNIT_ASSERT_VALUES_EQUAL(actual.Functions, expected.Functions);
    UNIT_ASSERT_VALUES_EQUAL(actual.Hints, expected.Hints);
}

Y_UNIT_TEST(FrequencyDataResouce) {
    TFrequencyData data = LoadFrequencyData();
    Y_UNUSED(data);
}

} // Y_UNIT_TEST_SUITE(FrequencyTests)