Add hope_list:map_result/2

[hope.git] / test / hope_list_SUITE.erl

diff --git a/test/hope_list_SUITE.erl b/test/hope_list_SUITE.erl
index 1bed138..7152503 100644 (file)
--- a/test/hope_list_SUITE.erl
+++ b/test/hope_list_SUITE.erl
@@ -15,6 +15,8 @@
     , t_map_rev/1
     , t_map_slow/1
     , t_map/1
+    , t_map_3/1
+    , t_map_result/1
     ]).
 
 
@@ -37,6 +39,8 @@ groups() ->
         , t_map_rev
         , t_map_slow
         , t_map
+        , t_map_3
+        , t_map_result
         ],
     Properties = [parallel],
     [{?GROUP, Properties, Tests}].
@@ -67,6 +71,13 @@ map() ->
     ?FORALL({L, F}, {list(integer()), function([integer()], term())},
             hope_list:map(L, F) == lists:map(F, L)).
 
+t_map_3(_Cfg) ->
+    ?PROPTEST(map_3).
+
+map_3() ->
+    ?FORALL({L, F, N}, {list(integer()), function([integer()], term()), non_neg_integer()},
+            hope_list:map(L, F, N) == lists:map(F, L)).
+
 t_unique_preserve_order(_Cfg) ->
     ?PROPTEST(prop_unique_preserve_order).
 
@@ -80,3 +91,13 @@ prop_unique_preserve_order() ->
 
 t_hope_list_specs(_) ->
     [] = proper:check_specs(hope_list).
+
+t_map_result(_Cfg) ->
+    AssertPositive =
+        fun (I) when I > 0 -> {ok, I}; (_) -> {error, negative} end,
+    AllPositives = lists:seq(1, 5),
+    AllNegatives = lists:seq(-5, -1),
+    Mixed = lists:seq(-5, 5),
+    {ok, AllPositives} = hope_list:map_result(AllPositives, AssertPositive),
+    {error, negative}  = hope_list:map_result(AllNegatives, AssertPositive),
+    {error, negative}  = hope_list:map_result(Mixed, AssertPositive).