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

more tests! yay!

Browse source
This commit is contained in:
Simon Cruanes 2015-03-17 01:01:06 +01:00
parent d4ccbccbb8
commit c1313e094d
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/data/CCFQueue.ml
View file

@ -133,6 +133,11 @@ let take_front_l n q =
aux (x::acc) q' (n-1) aux (x::acc) q' (n-1)
in aux [] q n in aux [] q n
(*$T
let l, q = take_front_l 5 (1 -- 10) in \
l = [1;2;3;4;5] && to_list q = [6;7;8;9;10]
*)
let take_front_while p q = let take_front_while p q =
let rec aux acc q = let rec aux acc q =
if is_empty q then List.rev acc, q if is_empty q then List.rev acc, q
@ -141,6 +146,10 @@ let take_front_while p q =
if p x then aux (x::acc) q' else List.rev acc, q if p x then aux (x::acc) q' else List.rev acc, q
in aux [] q in aux [] q
(*$T
take_front_while (fun x-> x<5) (1 -- 10) |> fst = [1;2;3;4]
*)
let rec take_back_exn : 'a. 'a t -> 'a t * 'a let rec take_back_exn : 'a. 'a t -> 'a t * 'a
= fun q -> match q with = fun q -> match q with
| Shallow Zero -> invalid_arg "FQueue.take_back_exn" | Shallow Zero -> invalid_arg "FQueue.take_back_exn"
@ -255,6 +264,13 @@ let nth i q =
try Some (nth_exn i q) try Some (nth_exn i q)
with Failure _ -> None with Failure _ -> None
(*$Q
(Q.list Q.int) (fun l -> \
let len = List.length l in let idx = CCList.(0 -- (len - 1)) in \
let q = of_list l in \
l = [] || List.for_all (fun i -> nth i q = Some (List.nth l i)) idx)
*)
let init q = let init q =
try fst (take_back_exn q) try fst (take_back_exn q)
with Empty -> q with Empty -> q