diff options
Diffstat (limited to 'gen')
| -rw-r--r-- | gen/src/map_dict.erl | 8 | ||||
| -rw-r--r-- | gen/test/map_dict_test.erl | 9 |
2 files changed, 15 insertions, 2 deletions
diff --git a/gen/src/map_dict.erl b/gen/src/map_dict.erl index 56fb686..eefdfc3 100644 --- a/gen/src/map_dict.erl +++ b/gen/src/map_dict.erl @@ -1,7 +1,7 @@ -module(map_dict). -compile(no_auto_import). --export([size/1, to_list/1, from_list/1, has_key/2, new/0, fetch/2, put/3, map_values/2, keys/1, values/1, filter/2]). +-export([size/1, to_list/1, from_list/1, has_key/2, new/0, fetch/2, put/3, map_values/2, keys/1, values/1, filter/2, take/2]). size(A) -> maps:size(A). @@ -47,3 +47,9 @@ erl_filter(A, B) -> filter(Map, Fun) -> erl_filter(Fun, Map). + +erl_take(A, B) -> + maps:with(A, B). + +take(Map, Keys) -> + erl_take(Keys, Map). diff --git a/gen/test/map_dict_test.erl b/gen/test/map_dict_test.erl index e4560d9..a5a0d8a 100644 --- a/gen/test/map_dict_test.erl +++ b/gen/test/map_dict_test.erl @@ -1,7 +1,7 @@ -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]). +-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]). from_list_test() -> expect:equal(map_dict:size(map_dict:from_list([{4, 0}, {1, 0}])), 2). @@ -53,3 +53,10 @@ values_test() -> {<<"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}])). |