mte/unikernel/duniverse/base/test/test_base_containers_mono.ml
2025-11-11 02:07:51 +01:00

142 lines
3.9 KiB
OCaml

open! Import
open Test_container
(* Tests of containers that are not polymorphic (i.e. have a fixed element type). *)
include (
Test_S0 (struct
include String
let mem t c = mem t c
module Elt = struct
type t = char [@@deriving sexp]
let of_int = Char.of_int_exn
let to_int = Char.to_int
end
let of_list = of_char_list
end) :
sig end)
let%expect_test "Hash_set" =
Base_container_tests.test_container_s0
(module struct
open Base_quickcheck
module Elt = struct
include Int
type t = (int[@generator Generator.small_strictly_positive_int])
[@@deriving compare, equal, quickcheck, sexp_of]
end
include Hash_set
type t = Hash_set.M(Int).t [@@deriving sexp_of]
let quickcheck_generator =
Generator.map [%generator: Elt.t list] ~f:(Hash_set.of_list (module Int))
;;
let quickcheck_observer = Observer.unmap [%observer: Elt.t list] ~f:Hash_set.to_list
let quickcheck_shrinker =
Shrinker.map
[%shrinker: Elt.t list]
~f:(Hash_set.of_list (module Int))
~f_inverse:Hash_set.to_list
;;
(* [to_list] and [to_array] proceed in the opposite order as everything else. This
is likely a performance hack to reuse [fold] without adding a [List.rev]. It is
not particularly problematic, since hash table order is already unpredictable due
to hash functions. *)
let to_list t = List.rev (to_list t)
let to_array t = Array.rev (to_array t)
end);
[%expect
{|
Container: testing [length]
Container: testing [is_empty]
Container: testing [mem]
Container: testing [iter]
Container: testing [fold]
Container: testing [fold_result]
Container: testing [fold_until]
Container: testing [exists]
Container: testing [for_all]
Container: testing [count]
Container: testing [sum]
Container: testing [find]
Container: testing [find_map]
Container: testing [to_list]
Container: testing [to_array]
Container: testing [min_elt]
Container: testing [max_elt]
|}]
;;
let%expect_test "String" =
Base_container_tests.test_indexed_container_s0_with_creators
(module struct
include String
module Elt = struct
type t = char [@@deriving compare, equal, quickcheck, sexp_of]
end
type t = string [@@deriving quickcheck]
(* eta-expand due to [local_] types *)
let mem t c = mem t c
(* leave off the [?sep] argument *)
let concat list = concat list
let concat_map list = concat_map list
let concat_mapi list = concat_mapi list
end);
[%expect
{|
Container: testing [length]
Container: testing [is_empty]
Container: testing [mem]
Container: testing [iter]
Container: testing [fold]
Container: testing [fold_result]
Container: testing [fold_until]
Container: testing [exists]
Container: testing [for_all]
Container: testing [count]
Container: testing [sum]
Container: testing [find]
Container: testing [find_map]
Container: testing [to_list]
Container: testing [to_array]
Container: testing [min_elt]
Container: testing [max_elt]
Container: testing [of_list]
Container: testing [of_array]
Container: testing [append]
Container: testing [concat]
Container: testing [map]
Container: testing [filter]
Container: testing [filter_map]
Container: testing [concat_map]
Container: testing [partition_tf]
Container: testing [partition_map]
Container: testing [foldi]
Container: testing [iteri]
Container: testing [existsi]
Container: testing [for_alli]
Container: testing [counti]
Container: testing [findi]
Container: testing [find_mapi]
Container: testing [init]
Container: testing [mapi]
Container: testing [filteri]
Container: testing [filter_mapi]
Container: testing [concat_mapi]
|}]
;;