-module(gleam@list_test).
-compile(no_auto_import).

-export([length_test/0, reverse_test/0, is_empty_test/0, contains_test/0, head_test/0, tail_test/0, filter_test/0, map_test/0, traverse_test/0, drop_test/0, take_test/0, new_test/0, append_test/0, flatten_test/0, fold_test/0, fold_right_test/0, find_map_test/0, find_test/0, all_test/0, any_test/0, zip_test/0, strict_zip_test/0, intersperse_test/0, at_test/0, unique_test/0, sort_test/0, index_map_test/0, range_test/0, repeat_test/0, split_test/0, split_while_test/0, key_find_test/0]).

length_test() ->
    gleam@should:equal(gleam@list:length([]), 0),
    gleam@should:equal(gleam@list:length([1]), 1),
    gleam@should:equal(gleam@list:length([1, 1]), 2),
    gleam@should:equal(gleam@list:length([1, 1, 1]), 3).

reverse_test() ->
    gleam@should:equal(gleam@list:reverse([]), []),
    gleam@should:equal(gleam@list:reverse([1, 2, 3, 4, 5]), [5, 4, 3, 2, 1]).

is_empty_test() ->
    gleam@should:be_true(gleam@list:is_empty([])),
    gleam@should:be_false(gleam@list:is_empty([1])).

contains_test() ->
    gleam@should:be_true(gleam@list:contains([0, 4, 5, 1], 1)),
    gleam@should:be_false(gleam@list:contains([0, 4, 5, 7], 1)),
    gleam@should:be_false(gleam@list:contains([], 1)).

head_test() ->
    gleam@should:equal(gleam@list:head([0, 4, 5, 7]), {ok, 0}),
    gleam@should:equal(gleam@list:head([]), {error, nil}).

tail_test() ->
    gleam@should:equal(gleam@list:tail([0, 4, 5, 7]), {ok, [4, 5, 7]}),
    gleam@should:equal(gleam@list:tail([0]), {ok, []}),
    gleam@should:equal(gleam@list:tail([]), {error, nil}).

filter_test() ->
    gleam@should:equal(gleam@list:filter([], fun(_) -> true end), []),
    gleam@should:equal(
        gleam@list:filter([0, 4, 5, 7, 3], fun(_) -> true end),
        [0, 4, 5, 7, 3]
    ),
    gleam@should:equal(
        gleam@list:filter([0, 4, 5, 7, 3], fun(X) -> X > 4 end),
        [5, 7]
    ),
    gleam@should:equal(
        gleam@list:filter([0, 4, 5, 7, 3], fun(X1) -> X1 < 4 end),
        [0, 3]
    ).

map_test() ->
    gleam@should:equal(gleam@list:map([], fun(X) -> X * 2 end), []),
    gleam@should:equal(
        gleam@list:map([0, 4, 5, 7, 3], fun(X1) -> X1 * 2 end),
        [0, 8, 10, 14, 6]
    ).

traverse_test() ->
    Fun = fun(X) -> case X =:= 6 orelse X =:= 5 orelse X =:= 4 of
            true ->
                {ok, X * 2};

            false ->
                {error, X}
        end end,
    gleam@should:equal(
        gleam@list:traverse([5, 6, 5, 6], Fun),
        {ok, [10, 12, 10, 12]}
    ),
    gleam@should:equal(gleam@list:traverse([4, 6, 5, 7, 3], Fun), {error, 7}).

drop_test() ->
    gleam@should:equal(gleam@list:drop([], 5), []),
    gleam@should:equal(gleam@list:drop([1, 2, 3, 4, 5, 6, 7, 8], 5), [6, 7, 8]).

take_test() ->
    gleam@should:equal(gleam@list:take([], 5), []),
    gleam@should:equal(
        gleam@list:take([1, 2, 3, 4, 5, 6, 7, 8], 5),
        [1, 2, 3, 4, 5]
    ).

new_test() ->
    gleam@should:equal(gleam@list:new(), []).

append_test() ->
    gleam@should:equal(gleam@list:append([1], [2, 3]), [1, 2, 3]).

flatten_test() ->
    gleam@should:equal(gleam@list:flatten([]), []),
    gleam@should:equal(gleam@list:flatten([[]]), []),
    gleam@should:equal(gleam@list:flatten([[], [], []]), []),
    gleam@should:equal(gleam@list:flatten([[1, 2], [], [3, 4]]), [1, 2, 3, 4]).

fold_test() ->
    gleam@should:equal(
        gleam@list:fold([1, 2, 3], [], fun(X, Acc) -> [X | Acc] end),
        [3, 2, 1]
    ).

fold_right_test() ->
    gleam@should:equal(
        gleam@list:fold_right([1, 2, 3], [], fun(X, Acc) -> [X | Acc] end),
        [1, 2, 3]
    ).

find_map_test() ->
    F = fun(X) -> case X of
            2 ->
                {ok, 4};

            _ ->
                {error, nil}
        end end,
    gleam@should:equal(gleam@list:find_map([1, 2, 3], F), {ok, 4}),
    gleam@should:equal(gleam@list:find_map([1, 3, 2], F), {ok, 4}),
    gleam@should:equal(gleam@list:find_map([1, 3], F), {error, nil}).

find_test() ->
    IsTwo = fun(X) -> X =:= 2 end,
    gleam@should:equal(gleam@list:find([1, 2, 3], IsTwo), {ok, 2}),
    gleam@should:equal(gleam@list:find([1, 3, 2], IsTwo), {ok, 2}),
    gleam@should:equal(gleam@list:find([1, 3], IsTwo), {error, nil}).

all_test() ->
    gleam@should:equal(
        gleam@list:all([1, 2, 3, 4, 5], fun(X) -> X > 0 end),
        true
    ),
    gleam@should:equal(
        gleam@list:all([1, 2, 3, 4, 5], fun(X1) -> X1 < 0 end),
        false
    ),
    gleam@should:equal(gleam@list:all([], fun(_) -> false end), true).

any_test() ->
    gleam@should:equal(
        gleam@list:any([1, 2, 3, 4, 5], fun(X) -> X =:= 2 end),
        true
    ),
    gleam@should:equal(
        gleam@list:any([1, 2, 3, 4, 5], fun(X1) -> X1 < 0 end),
        false
    ),
    gleam@should:equal(gleam@list:any([], fun(_) -> false end), false).

zip_test() ->
    gleam@should:equal(gleam@list:zip([], [1, 2, 3]), []),
    gleam@should:equal(gleam@list:zip([1, 2], []), []),
    gleam@should:equal(
        gleam@list:zip([1, 2, 3], [4, 5, 6]),
        [{1, 4}, {2, 5}, {3, 6}]
    ),
    gleam@should:equal(gleam@list:zip([5, 6], [1, 2, 3]), [{5, 1}, {6, 2}]),
    gleam@should:equal(gleam@list:zip([5, 6, 7], [1, 2]), [{5, 1}, {6, 2}]).

strict_zip_test() ->
    gleam@should:equal(
        gleam@list:strict_zip([], [1, 2, 3]),
        {error, length_mismatch}
    ),
    gleam@should:equal(
        gleam@list:strict_zip([1, 2], []),
        {error, length_mismatch}
    ),
    gleam@should:equal(
        gleam@list:strict_zip([1, 2, 3], [4, 5, 6]),
        {ok, [{1, 4}, {2, 5}, {3, 6}]}
    ),
    gleam@should:equal(
        gleam@list:strict_zip([5, 6], [1, 2, 3]),
        {error, length_mismatch}
    ),
    gleam@should:equal(
        gleam@list:strict_zip([5, 6, 7], [1, 2]),
        {error, length_mismatch}
    ).

intersperse_test() ->
    gleam@should:equal(gleam@list:intersperse([1, 2, 3], 4), [1, 4, 2, 4, 3]),
    gleam@should:equal(gleam@list:intersperse([], 2), []).

at_test() ->
    gleam@should:equal(gleam@list:at([1, 2, 3], 2), {ok, 3}),
    gleam@should:equal(gleam@list:at([1, 2, 3], 5), {error, nil}),
    gleam@should:equal(gleam@list:at([], 0), {error, nil}),
    gleam@should:equal(gleam@list:at([1, 2, 3, 4, 5, 6], -1), {error, nil}).

unique_test() ->
    gleam@should:equal(
        gleam@list:unique([1, 1, 2, 3, 4, 4, 4, 5, 6]),
        [1, 2, 3, 4, 5, 6]
    ),
    gleam@should:equal(
        gleam@list:unique([7, 1, 45, 6, 2, 47, 2, 7, 5]),
        [7, 1, 45, 6, 2, 47, 5]
    ),
    gleam@should:equal(gleam@list:unique([3, 4, 5]), [3, 4, 5]),
    gleam@should:equal(gleam@list:unique([]), []).

sort_test() ->
    gleam@should:equal(
        gleam@list:sort([4, 3, 6, 5, 4], fun gleam@int:compare/2),
        [3, 4, 4, 5, 6]
    ),
    gleam@should:equal(
        gleam@list:sort([4, 3, 6, 5, 4, 1], fun gleam@int:compare/2),
        [1, 3, 4, 4, 5, 6]
    ),
    gleam@should:equal(
        gleam@list:sort([4.1, 3.1, 6.1, 5.1, 4.1], fun gleam@float:compare/2),
        [3.1, 4.1, 4.1, 5.1, 6.1]
    ),
    gleam@should:equal(gleam@list:sort([], fun gleam@int:compare/2), []).

index_map_test() ->
    gleam@should:equal(
        gleam@list:index_map([3, 4, 5], fun(I, X) -> {I, X} end),
        [{0, 3}, {1, 4}, {2, 5}]
    ),
    F = fun(I1, X1) -> gleam@string:append(X1, gleam@int:to_string(I1)) end,
    gleam@should:equal(
        gleam@list:index_map([<<"a">>, <<"b">>, <<"c">>], F),
        [<<"a0">>, <<"b1">>, <<"c2">>]
    ).

range_test() ->
    gleam@should:equal(gleam@list:range(0, 0), []),
    gleam@should:equal(gleam@list:range(1, 1), []),
    gleam@should:equal(gleam@list:range(-1, -1), []),
    gleam@should:equal(gleam@list:range(0, 1), [0]),
    gleam@should:equal(gleam@list:range(0, 5), [0, 1, 2, 3, 4]),
    gleam@should:equal(gleam@list:range(1, -5), [1, 0, -1, -2, -3, -4]).

repeat_test() ->
    gleam@should:equal(gleam@list:repeat(1, -10), []),
    gleam@should:equal(gleam@list:repeat(1, 0), []),
    gleam@should:equal(gleam@list:repeat(2, 3), [2, 2, 2]),
    gleam@should:equal(
        gleam@list:repeat(<<"x">>, 5),
        [<<"x">>, <<"x">>, <<"x">>, <<"x">>, <<"x">>]
    ).

split_test() ->
    gleam@should:equal(gleam@list:split([], 0), {[], []}),
    gleam@should:equal(
        gleam@list:split([0, 1, 2, 3, 4], 0),
        {[], [0, 1, 2, 3, 4]}
    ),
    gleam@should:equal(
        gleam@list:split([0, 1, 2, 3, 4], -2),
        {[], [0, 1, 2, 3, 4]}
    ),
    gleam@should:equal(
        gleam@list:split([0, 1, 2, 3, 4], 1),
        {[0], [1, 2, 3, 4]}
    ),
    gleam@should:equal(
        gleam@list:split([0, 1, 2, 3, 4], 3),
        {[0, 1, 2], [3, 4]}
    ),
    gleam@should:equal(
        gleam@list:split([0, 1, 2, 3, 4], 9),
        {[0, 1, 2, 3, 4], []}
    ).

split_while_test() ->
    gleam@should:equal(
        gleam@list:split_while([], fun(X) -> X =< 5 end),
        {[], []}
    ),
    gleam@should:equal(
        gleam@list:split_while([1, 2, 3, 4, 5], fun(X1) -> X1 =< 5 end),
        {[1, 2, 3, 4, 5], []}
    ),
    gleam@should:equal(
        gleam@list:split_while([1, 2, 3, 4, 5], fun(X2) -> X2 =:= 2 end),
        {[], [1, 2, 3, 4, 5]}
    ),
    gleam@should:equal(
        gleam@list:split_while([1, 2, 3, 4, 5], fun(X3) -> X3 =< 3 end),
        {[1, 2, 3], [4, 5]}
    ),
    gleam@should:equal(
        gleam@list:split_while([1, 2, 3, 4, 5], fun(X4) -> X4 =< -3 end),
        {[], [1, 2, 3, 4, 5]}
    ).

key_find_test() ->
    Proplist = [{0, <<"1">>}, {1, <<"2">>}],
    gleam@should:equal(gleam@list:key_find(Proplist, 0), {ok, <<"1">>}),
    gleam@should:equal(gleam@list:key_find(Proplist, 1), {ok, <<"2">>}),
    gleam@should:equal(gleam@list:key_find(Proplist, 2), {error, nil}).