Current section

Files

Jump to
gleam_stdlib gen test gleam@function_test.erl
Raw

gen/test/gleam@function_test.erl

-module(gleam@function_test).
-compile(no_auto_import).
-export([compose_test/0, curry2_test/0, curry3_test/0, curry4_test/0, curry5_test/0, curry6_test/0, flip_test/0, identity_test/0, rescue_test/0]).
compose_test() ->
AddTwo = fun(Int) -> Int + 2 end,
AddThree = fun(Int@1) -> Int@1 + 3 end,
AddFive = gleam@function:compose(AddTwo, AddThree),
gleam@should:equal(AddFive(1), 6),
HeadToString = gleam@function:compose(
gleam@function:compose(
fun gleam@list:head/1,
fun(GleamCaptureVariable) ->
gleam@result:unwrap(GleamCaptureVariable, 0)
end
),
fun gleam@int:to_string/1
),
gleam@should:equal(HeadToString([1]), <<"1"/utf8>>),
gleam@should:equal(HeadToString([]), <<"0"/utf8>>).
curry2_test() ->
Fun = fun(A, B) -> A + B end,
Curried = gleam@function:curry2(Fun),
gleam@should:equal((Curried(1))(2), 3).
curry3_test() ->
Fun = fun(A, B, C) -> (A + B) + C end,
Curried = gleam@function:curry3(Fun),
gleam@should:equal(((Curried(1))(2))(4), 7).
curry4_test() ->
Fun = fun(A, B, C, D) -> ((A + B) + C) + D end,
Curried = gleam@function:curry4(Fun),
gleam@should:equal((((Curried(1))(2))(4))(8), 15).
curry5_test() ->
Fun = fun(A, B, C, D, E) -> (((A + B) + C) + D) + E end,
Curried = gleam@function:curry5(Fun),
gleam@should:equal(((((Curried(1))(2))(4))(8))(16), 31).
curry6_test() ->
Fun = fun(A, B, C, D, E, F) -> ((((A + B) + C) + D) + E) + F end,
Curried = gleam@function:curry6(Fun),
gleam@should:equal((((((Curried(1))(2))(4))(8))(16))(32), 63).
flip_test() ->
Fun = fun(S, I) ->
gleam@string:append(
gleam@string:append(
gleam@string:append(
gleam@string:append(<<"String: '"/utf8>>, S),
<<"', Int: '"/utf8>>
),
gleam@int:to_string(I)
),
<<"'"/utf8>>
)
end,
FlippedFun = gleam@function:flip(Fun),
gleam@should:equal(
Fun(<<"Bob"/utf8>>, 1),
<<"String: 'Bob', Int: '1'"/utf8>>
),
gleam@should:equal(
FlippedFun(2, <<"Alice"/utf8>>),
<<"String: 'Alice', Int: '2'"/utf8>>
).
identity_test() ->
gleam@should:equal(gleam@function:identity(1), 1),
gleam@should:equal(gleam@function:identity(<<""/utf8>>), <<""/utf8>>),
gleam@should:equal(gleam@function:identity([]), []),
gleam@should:equal(gleam@function:identity({1, 2.0}), {1, 2.0}).
rescue_test() ->
gleam@should:equal(gleam@function:rescue(fun() -> 1 end), {ok, 1}),
gleam@should:equal(
gleam@function:rescue(fun() -> erlang:throw(1) end),
{error, {thrown, gleam@dynamic:from(1)}}
),
gleam@should:equal(
gleam@function:rescue(fun() -> erlang:error(<<""/utf8>>) end),
{error, {errored, gleam@dynamic:from(<<""/utf8>>)}}
).