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 eefdfc3..7d97a7d 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, take/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, drop/2]). size(A) -> maps:size(A). @@ -53,3 +53,9 @@ erl_take(A, B) -> take(Map, Keys) -> erl_take(Keys, Map). + +erl_drop(A, B) -> + maps:without(A, B). + +drop(Map, Keys) -> + erl_drop(Keys, Map). diff --git a/gen/test/map_dict_test.erl b/gen/test/map_dict_test.erl index a5a0d8a..e59911e 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, take_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, drop_test/0]). from_list_test() -> expect:equal(map_dict:size(map_dict:from_list([{4, 0}, {1, 0}])), 2). @@ -60,3 +60,10 @@ take_test() -> {<<"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}])). |