diff options
Diffstat (limited to 'gen')
| -rw-r--r-- | gen/src/gleam@pair.erl | 10 | ||||
| -rw-r--r-- | gen/test/gleam@pair_test.erl | 16 |
2 files changed, 24 insertions, 2 deletions
diff --git a/gen/src/gleam@pair.erl b/gen/src/gleam@pair.erl index 887adb9..3361dc7 100644 --- a/gen/src/gleam@pair.erl +++ b/gen/src/gleam@pair.erl @@ -1,7 +1,7 @@ -module(gleam@pair). -compile(no_auto_import). --export([first/1, second/1, swap/1]). +-export([first/1, second/1, swap/1, map_first/2, map_second/2]). first(Tup) -> {A, _} = Tup, @@ -14,3 +14,11 @@ second(Tup) -> swap(Tup) -> {A, B} = Tup, {B, A}. + +map_first(Tup, F) -> + {A, B} = Tup, + {F(A), B}. + +map_second(Tup, F) -> + {A, B} = Tup, + {A, F(B)}. diff --git a/gen/test/gleam@pair_test.erl b/gen/test/gleam@pair_test.erl index 775f161..6a2c856 100644 --- a/gen/test/gleam@pair_test.erl +++ b/gen/test/gleam@pair_test.erl @@ -1,7 +1,7 @@ -module(gleam@pair_test). -compile(no_auto_import). --export([first_test/0, second_test/0, swap_test/0]). +-export([first_test/0, second_test/0, swap_test/0, map_first_test/0, map_second_test/0]). first_test() -> gleam@expect:equal(gleam@pair:first({1, 2}), 1), @@ -13,3 +13,17 @@ second_test() -> swap_test() -> gleam@expect:equal(gleam@pair:swap({1, <<"2">>}), {<<"2">>, 1}). + +map_first_test() -> + Inc = fun(A) -> A + 1 end, + gleam@expect:equal(gleam@pair:map_first({1, 2}, Inc), {2, 2}), + gleam@expect:equal(gleam@pair:map_first({8, 2}, Inc), {9, 2}), + gleam@expect:equal(gleam@pair:map_first({0, -2}, Inc), {1, -2}), + gleam@expect:equal(gleam@pair:map_first({-10, 20}, Inc), {-9, 20}). + +map_second_test() -> + Dec = fun(A) -> A - 1 end, + gleam@expect:equal(gleam@pair:map_second({1, 2}, Dec), {1, 1}), + gleam@expect:equal(gleam@pair:map_second({8, 2}, Dec), {8, 1}), + gleam@expect:equal(gleam@pair:map_second({0, -2}, Dec), {0, -3}), + gleam@expect:equal(gleam@pair:map_second({-10, 20}, Dec), {-10, 19}). |