aboutsummaryrefslogtreecommitdiff
path: root/gen/test/map_dict_test.erl
blob: 03be65a355b2b5a6ce2345293c6842f2f30fd283 (plain)
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
-module(map_dict_test).
-compile(no_auto_import).

-export([from_list_test/0, has_key_test/0, new_test/0, fetch_test/0, put_test/0, map_values_test/0, keys_test/0, values_test/0, take_test/0, drop_test/0, merge_test/0, delete_test/0, update_test/0, fold_test/0]).

from_list_test() ->
    expect:equal(map_dict:size(map_dict:from_list([{4, 0}, {1, 0}])), 2).

has_key_test() ->
    expect:false(map_dict:has_key(map_dict:from_list([]), 1)),
    expect:true(map_dict:has_key(map_dict:from_list([{1, 0}]), 1)),
    expect:true(map_dict:has_key(map_dict:from_list([{4, 0}, {1, 0}]), 1)),
    expect:false(map_dict:has_key(map_dict:from_list([{4, 0}, {1, 0}]), 0)).

new_test() ->
    expect:equal(map_dict:size(map_dict:new()), 0),
    expect:equal(map_dict:to_list(map_dict:new()), []).

fetch_test() ->
    Proplist = [{4, 0}, {1, 1}],
    M = map_dict:from_list(Proplist),
    expect:equal(map_dict:fetch(M, 4), {ok, 0}),
    expect:equal(map_dict:fetch(M, 1), {ok, 1}),
    expect:is_error(map_dict:fetch(M, 2)).

put_test() ->
    expect:equal(map_dict:put(map_dict:put(map_dict:put(map_dict:new(),
                                                        <<"a">>,
                                                        0),
                                           <<"b">>,
                                           1),
                              <<"c">>,
                              2),
                 map_dict:from_list([{<<"a">>, 0},
                                     {<<"b">>, 1},
                                     {<<"c">>, 2}])).

map_values_test() ->
    expect:equal(map_dict:map_values(map_dict:from_list([{1, 0},
                                                         {2, 1},
                                                         {3, 2}]),
                                     fun(K, V) -> K + V end),
                 map_dict:from_list([{1, 1}, {2, 3}, {3, 5}])).

keys_test() ->
    expect:equal(map_dict:keys(map_dict:from_list([{<<"a">>, 0},
                                                   {<<"b">>, 1},
                                                   {<<"c">>, 2}])),
                 [<<"a">>, <<"b">>, <<"c">>]).

values_test() ->
    expect:equal(map_dict:values(map_dict:from_list([{<<"a">>, 0},
                                                     {<<"b">>, 1},
                                                     {<<"c">>, 2}])),
                 [0, 1, 2]).

take_test() ->
    expect:equal(map_dict:take(map_dict:from_list([{<<"a">>, 0},
                                                   {<<"b">>, 1},
                                                   {<<"c">>, 2}]),
                               [<<"a">>, <<"b">>, <<"d">>]),
                 map_dict:from_list([{<<"a">>, 0}, {<<"b">>, 1}])).

drop_test() ->
    expect:equal(map_dict:drop(map_dict:from_list([{<<"a">>, 0},
                                                   {<<"b">>, 1},
                                                   {<<"c">>, 2}]),
                               [<<"a">>, <<"b">>, <<"d">>]),
                 map_dict:from_list([{<<"c">>, 2}])).

merge_test() ->
    A = map_dict:from_list([{<<"a">>, 2}, {<<"c">>, 4}, {<<"d">>, 3}]),
    B = map_dict:from_list([{<<"a">>, 0}, {<<"b">>, 1}, {<<"c">>, 2}]),
    expect:equal(map_dict:merge(A, B),
                 map_dict:from_list([{<<"a">>, 0},
                                     {<<"b">>, 1},
                                     {<<"c">>, 2},
                                     {<<"d">>, 3}])),
    expect:equal(map_dict:merge(B, A),
                 map_dict:from_list([{<<"a">>, 2},
                                     {<<"b">>, 1},
                                     {<<"c">>, 4},
                                     {<<"d">>, 3}])).

delete_test() ->
    expect:equal(map_dict:delete(map_dict:delete(map_dict:from_list([{<<"a">>,
                                                                      0},
                                                                     {<<"b">>,
                                                                      1},
                                                                     {<<"c">>,
                                                                      2}]),
                                                 <<"a">>),
                                 <<"d">>),
                 map_dict:from_list([{<<"b">>, 1}, {<<"c">>, 2}])).

update_test() ->
    Dict = map_dict:from_list([{<<"a">>, 0}, {<<"b">>, 1}, {<<"c">>, 2}]),
    IncOrZero = fun(X) -> case X of
            {ok, I} ->
                I + 1;

            {error, _} ->
                0
        end end,
    expect:equal(map_dict:update(Dict, <<"a">>, IncOrZero),
                 map_dict:from_list([{<<"a">>, 1},
                                     {<<"b">>, 1},
                                     {<<"c">>, 2}])),
    expect:equal(map_dict:update(Dict, <<"b">>, IncOrZero),
                 map_dict:from_list([{<<"a">>, 0},
                                     {<<"b">>, 2},
                                     {<<"c">>, 2}])),
    expect:equal(map_dict:update(Dict, <<"z">>, IncOrZero),
                 map_dict:from_list([{<<"a">>, 0},
                                     {<<"b">>, 1},
                                     {<<"c">>, 2},
                                     {<<"z">>, 0}])).

fold_test() ->
    Dict = map_dict:from_list([{<<"a">>, 0},
                               {<<"b">>, 1},
                               {<<"c">>, 2},
                               {<<"d">>, 3}]),
    Add = fun(_, V, Acc) -> V + Acc end,
    expect:equal(map_dict:fold(Dict, 0, Add), 6),
    Concat = fun(K, _, Acc) -> str:append(Acc, K) end,
    expect:equal(map_dict:fold(Dict, <<"">>, Concat), <<"abcd">>),
    expect:equal(map_dict:fold(map_dict:from_list([]), 0, Add), 0).