simon/iter
1
0
Fork 0
mirror of https://github.com/c-cube/iter.git synced 2025-12-07 03:35:32 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

update tests

Browse source
This commit is contained in:
Simon Cruanes 2013-02-06 14:53:47 +01:00
parent 2bf5b628ab
commit f0c3d399bc
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
tests.ml
View file

@ -3,48 +3,48 @@
(** print a list of items using the printing function *) (** print a list of items using the printing function *)
let pp_list ?(sep=", ") pp_item formatter l = let pp_list ?(sep=", ") pp_item formatter l =
Sequence.pp_seq ~sep pp_item formatter (Sequence.List.to_seq l) Sequence.pp_seq ~sep pp_item formatter (Sequence.of_list l)
(** Set of integers *) (** Set of integers *)
module ISet = Set.Make(struct type t = int let compare = compare end) module ISet = Set.Make(struct type t = int let compare = compare end)
module ISetSeq = Sequence.Set(ISet) let iset = (module ISet : Set.S with type elt = int and type t = ISet.t)
let sexpr = "(foo bar (bazz quux hello 42) world (zoo foo bar (1 2 (3 4))))" let sexpr = "(foo bar (bazz quux hello 42) world (zoo foo bar (1 2 (3 4))))"
let _ = let _ =
let l = [0;1;2;3;4;5;6] in let l = [0;1;2;3;4;5;6] in
let l' = Sequence.List.of_seq let l' = Sequence.to_list
(Sequence.filter (fun x -> x mod 2 = 0) (Sequence.List.to_seq l)) in (Sequence.filter (fun x -> x mod 2 = 0) (Sequence.of_list l)) in
let l'' = Sequence.List.of_seq let l'' = Sequence.to_list
(Sequence.take 3 (Sequence.drop 1 (Sequence.List.to_seq l))) in (Sequence.take 3 (Sequence.drop 1 (Sequence.of_list l))) in
let h = Hashtbl.create 3 in let h = Hashtbl.create 3 in
for i = 0 to 5 do for i = 0 to 5 do
Hashtbl.add h i (i*i); Hashtbl.add h i (i*i);
done; done;
let l2 = Sequence.List.of_seq let l2 = Sequence.to_list
(Sequence.map (fun (x, y) -> (string_of_int x) ^ " -> " ^ (string_of_int y)) (Sequence.map (fun (x, y) -> (string_of_int x) ^ " -> " ^ (string_of_int y))
(Sequence.Hashtbl.to_seq h)) (Sequence.of_hashtbl h))
in in
let l3 = Sequence.List.of_seq (Sequence.rev (Sequence.Int.range ~start:0 ~stop:42)) in let l3 = Sequence.to_list (Sequence.rev (Sequence.int_range ~start:0 ~stop:42)) in
let set = List.fold_left (fun set x -> ISet.add x set) ISet.empty [4;3;100;42] in let set = List.fold_left (fun set x -> ISet.add x set) ISet.empty [4;3;100;42] in
let l4 = Sequence.List.of_seq (ISetSeq.to_seq set) in let l4 = Sequence.to_list (Sequence.of_set iset set) in
Format.printf "l=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l; Format.printf "l=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l;
Format.printf "l'=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l'; Format.printf "l'=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l';
Format.printf "l''=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l''; Format.printf "l''=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l'';
Format.printf "l2=@[<h>[%a]@]@." (pp_list Format.pp_print_string) l2; Format.printf "l2=@[<h>[%a]@]@." (pp_list Format.pp_print_string) l2;
Format.printf "l3=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l3; Format.printf "l3=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l3;
Format.printf "s={@[<h>%a@]}@." (Sequence.pp_seq Format.pp_print_int) (ISetSeq.to_seq set); Format.printf "s={@[<h>%a@]}@." (Sequence.pp_seq Format.pp_print_int) (Sequence.of_set iset set);
Format.printf "l4=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l4; Format.printf "l4=@[<h>[%a]@]@." (pp_list Format.pp_print_int) l4;
Format.printf "l3[:5]+l4=@[<h>[%a]@]@." (Sequence.pp_seq Format.pp_print_int) Format.printf "l3[:5]+l4=@[<h>[%a]@]@." (Sequence.pp_seq Format.pp_print_int)
(Sequence.Array.to_seq (Sequence.of_array
(Sequence.Array.of_seq (Sequence.append (Sequence.to_array (Sequence.append
(Sequence.take 5 (Sequence.List.to_seq l3)) (Sequence.List.to_seq l4)))); (Sequence.take 5 (Sequence.of_list l3)) (Sequence.of_list l4))));
(* sum *) (* sum *)
let n = 100000000 in let n = 100000000 in
let sum = Sequence.fold (+) 0 (Sequence.take n (Sequence.repeat 1)) in let sum = Sequence.fold (+) 0 (Sequence.take n (Sequence.repeat 1)) in
Format.printf "%dx1 = %d@." n sum; Format.printf "%dx1 = %d@." n sum;
assert (n=sum); assert (n=sum);
let s = Sexpr.of_seq (Sexpr.lex (Sequence.String.to_seq sexpr)) in let s = Sexpr.of_seq (Sexpr.lex (Sequence.of_str sexpr)) in
let s = Sexpr.of_seq (Sequence.map let s = Sexpr.of_seq (Sequence.map
(function | `Atom s -> `Atom (String.capitalize s) | tok -> tok) (function | `Atom s -> `Atom (String.capitalize s) | tok -> tok)
(Sexpr.traverse s)) (Sexpr.traverse s))