unikernel/duniverse/base/test/test_base_containers_mono.ml

@@ -0,0 +1,142 @@
+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]
+    |}]
+;;