simon/ocaml-containers
1
0
Fork 0
mirror of https://github.com/c-cube/ocaml-containers.git synced 2025-12-06 03:05:28 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

more tests for Enum

Browse source
This commit is contained in:
Simon Cruanes 2013-03-19 11:35:48 +01:00
parent 96d3c7e8b7
commit 4d293209a5
1 changed files with 65 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
tests/test_enum.ml
View file

@ -40,6 +40,63 @@ let test_flatMap () =
OUnit.assert_equal [1;2;2;3;3;4] (Enum.to_list e');
()
let test_zip () =
let e = Enum.zipWith (+) (Enum.repeat 1) (4--7) in
OUnit.assert_equal [5;6;7;8] (Enum.to_list e);
()
let test_filterMap () =
let f x = if x mod 2 = 0 then Some (string_of_int x) else None in
let e = Enum.filterMap f (1 -- 10) in
OUnit.assert_equal ["2"; "4"; "6"; "8"; "10"] (Enum.to_list e);
()
let test_round_robin () =
let e = Enum.of_list [1--3; 4--6; 7--9] in
let e' = Enum.round_robin e in
OUnit.assert_equal [1;4;7;2;5;8;3;6;9] (Enum.to_list e');
()
let test_persistent () =
let i = ref 0 in
let gen () =
let j = !i in
if j > 5 then raise Enum.EOG else (incr i; j)
in
let e = Enum.persistent gen in
OUnit.assert_equal [0;1;2;3;4;5] (Enum.to_list e);
OUnit.assert_equal [0;1;2;3;4;5] (Enum.to_list e);
OUnit.assert_equal [0;1;2;3;4;5] (Enum.to_list e);
()
let test_tee () =
let e = Enum.tee ~n:2 (1--10) in
let e = Enum.map Enum.persistent e in
let l = Enum.to_list e in
match l with
| [a;b] ->
OUnit.assert_equal [1;3;5;7;9] (Enum.to_list a);
OUnit.assert_equal [2;4;6;8;10] (Enum.to_list b)
| _ -> OUnit.assert_failure "wrong list lenght"
let test_interleave () =
let e1 = Enum.of_list [1;3;5;7;9] in
let e2 = Enum.of_list [2;4;6;8;10] in
let e = Enum.interleave e1 e2 in
OUnit.assert_equal [1;2;3;4;5;6;7;8;9;10] (Enum.to_list e);
()
let test_intersperse () =
let e = 1 -- 5 in
let e' = Enum.intersperse 0 e in
OUnit.assert_equal [1;0;2;0;3;0;4;0;5] (Enum.to_list e');
()
let test_product () =
let e = Enum.product (1--3) (4--5) in
OUnit.assert_equal [1,4; 1,5; 2,4; 2,5; 3,4; 3,5] (Enum.to_list e);
()
let suite =
"test_enum" >:::
[ "test_singleton" >:: test_singleton;
@ -47,4 +104,12 @@ let suite =
"test_map" >:: test_map;
"test_append" >:: test_append;
"test_flatMap" >:: test_flatMap;
"test_zip" >:: test_zip;
"test_filterMap" >:: test_filterMap;
"test_round_robin" >:: test_round_robin;
"test_persistent" >:: test_persistent;
"test_tee" >:: test_tee;
"test_interleave" >:: test_interleave;
"test_intersperse" >:: test_intersperse;
"test_product" >:: test_product;
]