b6449cae87
* `lib_stdlib`: basic extended OCaml stdlib and generic data structures
* `lib_data_encoding`: almost independant 'Data_encoding'
* `lib_error_monad`: almost independant 'Error_monad'
* `lib_stdlib_lwt`: extended Lwt library
* `lib_crypto`: all the crypto stuff (hashing, signing, cryptobox).
* `lib_base`:
- basic type definitions (Block_header, Operation, ...)
- a module `TzPervasives` to bind them all and to be the
single module opened everywhere.
In the process, I splitted `Tezos_data` and `Hash` in multiple
submodules, thus removing a lot of `-open`.
The following two modules may not have found their place yet:
- Base58 (currently in `lib_crypto`)
- Cli_entries (currently in `lib_stdlib_lwt`)
54 lines
2.1 KiB
OCaml
54 lines
2.1 KiB
OCaml
(**************************************************************************)
|
|
(* *)
|
|
(* Copyright (c) 2014 - 2017. *)
|
|
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
|
|
(* *)
|
|
(* All rights reserved. No warranty, explicit or implicit, provided. *)
|
|
(* *)
|
|
(**************************************************************************)
|
|
|
|
let rec permut = function
|
|
| [] -> [[]]
|
|
| x :: xs ->
|
|
let insert xs =
|
|
let rec loop acc left right =
|
|
match right with
|
|
| [] -> List.rev (x :: left) :: acc
|
|
| y :: ys ->
|
|
loop
|
|
((List.rev_append left (x :: right)) :: acc)
|
|
(y :: left) ys in
|
|
loop [] [] xs in
|
|
List.concat (List.map insert (permut xs))
|
|
|
|
let test_take_n _ =
|
|
ListLabels.iter (permut [1;2;3;4;5;6;7;8;9]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare 1 xs) [9]
|
|
end ;
|
|
ListLabels.iter (permut [1;2;3;4;5;6;7;8;9]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare 3 xs) [7;8;9]
|
|
end ;
|
|
let inv_compare x y = compare y x in
|
|
ListLabels.iter (permut [1;2;3;4;5;6;7;8;9]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare:inv_compare 3 xs) [3;2;1]
|
|
end ;
|
|
(* less elements than the bound. *)
|
|
ListLabels.iter (permut [1;2;3;4;5;6;7;8;9]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare 12 xs) [1;2;3;4;5;6;7;8;9]
|
|
end ;
|
|
(* with duplicates. *)
|
|
ListLabels.iter (permut [1;2;3;3;4;5;5;5;6]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare 3 xs) [5;5;6]
|
|
end ;
|
|
ListLabels.iter (permut [1;2;3;3;4;5;5;5;6]) ~f:begin fun xs ->
|
|
Assert.equal ~msg:__LOC__ (List.take_n ~compare 5 xs) [4;5;5;5;6]
|
|
end ;
|
|
return ()
|
|
|
|
let tests : (string * (string -> unit tzresult Lwt.t)) list = [
|
|
"take_n", test_take_n ;
|
|
]
|
|
|
|
let () =
|
|
Test.run "utils." tests
|