simon/iter
1
0
Fork 0
mirror of https://github.com/c-cube/iter.git synced 2025-12-06 19:25:30 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

bugfix: take would fail on 0

Browse source
This commit is contained in:
Simon Cruanes 2014-08-08 23:47:51 +02:00
parent 4ae9e77db3
commit 81d7315000
2 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
sequence.ml
View file

@ -336,9 +336,9 @@ let take n seq k =
let count = ref 0 in let count = ref 0 in
try try
seq (fun x -> seq (fun x ->
if !count = n then raise ExitSequence;
incr count; incr count;
k x; k x;
if !count = n then raise ExitSequence
) )
with ExitSequence -> () with ExitSequence -> ()

8
tests/test_sequence.ml
View file

@ -190,6 +190,13 @@ let test_int_range () =
OUnit.assert_equal ~printer:pp_ilist [] S.(to_list (10 --^ 60)); OUnit.assert_equal ~printer:pp_ilist [] S.(to_list (10 --^ 60));
() ()
let test_take () =
let l = S.(to_list (take 0 (of_list [1]))) in
OUnit.assert_equal ~printer:pp_ilist [] l;
let l = S.(to_list (take 5 (of_list [1;2;3;4;5;6;7;8;9;10]))) in
OUnit.assert_equal ~printer:pp_ilist [1;2;3;4;5] l;
()
let suite = let suite =
"test_sequence" >::: "test_sequence" >:::
[ "test_empty" >:: test_empty; [ "test_empty" >:: test_empty;
@ -217,4 +224,5 @@ let suite =
"test_unfoldr" >:: test_unfoldr; "test_unfoldr" >:: test_unfoldr;
"test_hashtbl" >:: test_hashtbl; "test_hashtbl" >:: test_hashtbl;
"test_int_range" >:: test_int_range; "test_int_range" >:: test_int_range;
"test_take" >:: test_take;
] ]