module Test = (val Containers_testlib.make ~__FILE__ ())
open Test

module Mixmap = struct
  t @@ fun () ->
  let module M = CCMixmap.Make (CCInt) in
  let inj_int = CCMixmap.create_inj () in
  let inj_str = CCMixmap.create_inj () in
  let inj_list_int = CCMixmap.create_inj () in

  let m =
    M.empty |> M.add ~inj:inj_int 1 1 |> M.add ~inj:inj_str 2 "2"
    |> M.add ~inj:inj_list_int 3 [ 3; 3; 3 ]
  in

  assert_equal (M.get ~inj:inj_int 1 m) (Some 1);
  assert_equal (M.get ~inj:inj_str 1 m) None;
  assert_equal (M.get ~inj:inj_str 2 m) (Some "2");
  assert_equal (M.get ~inj:inj_int 2 m) None;
  assert_equal (M.get ~inj:inj_list_int 3 m) (Some [ 3; 3; 3 ]);
  assert_equal (M.get ~inj:inj_str 3 m) None;
  true
end

module Mixset = struct
  open CCMixset;;

  t @@ fun () ->
  let k1 : int key = newkey () in
  let k2 : int key = newkey () in
  let k3 : string key = newkey () in
  let set = empty |> set ~key:k1 1 |> set ~key:k2 2 |> set ~key:k3 "3" in
  assert (get ~key:k1 set = Some 1);
  assert (get ~key:k2 set = Some 2);
  assert (get ~key:k3 set = Some "3");
  true
end

module Mixtbl = struct
  open CCFun
  open CCMixtbl;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let tbl = create 10 in
  assert_equal None (get ~inj:inj_int tbl "a");
  set ~inj:inj_int tbl "a" 1;
  assert_equal (Some 1) (get ~inj:inj_int tbl "a");
  let inj_string = create_inj () in
  set ~inj:inj_string tbl "b" "Hello";
  assert_equal (Some "Hello") (get ~inj:inj_string tbl "b");
  assert_equal None (get ~inj:inj_string tbl "a");
  assert_equal (Some 1) (get ~inj:inj_int tbl "a");
  set ~inj:inj_string tbl "a" "Bye";
  assert_equal None (get ~inj:inj_int tbl "a");
  assert_equal (Some "Bye") (get ~inj:inj_string tbl "a");
  true
  ;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let tbl = create 5 in
  set ~inj:inj_int tbl "foo" 1;
  set ~inj:inj_int tbl "bar" 2;
  assert_equal 2 (length tbl);
  assert_equal 2 (find ~inj:inj_int tbl "bar");
  set ~inj:inj_int tbl "foo" 42;
  assert_equal 2 (length tbl);
  remove tbl "bar";
  assert_equal 1 (length tbl);
  true
  ;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let inj_str = create_inj () in
  let tbl = create 5 in
  set ~inj:inj_int tbl "foo" 1;
  set ~inj:inj_int tbl "bar" 2;
  set ~inj:inj_str tbl "baaz" "hello";
  assert_equal 3 (length tbl);
  clear tbl;
  assert_equal 0 (length tbl);
  true
  ;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let inj_str = create_inj () in
  let tbl = create 5 in
  set ~inj:inj_int tbl "foo" 1;
  set ~inj:inj_int tbl "bar" 2;
  set ~inj:inj_str tbl "baaz" "hello";
  assert_bool "mem foo int" (mem ~inj:inj_int tbl "foo");
  assert_bool "mem bar int" (mem ~inj:inj_int tbl "bar");
  assert_bool "not mem baaz int" (not (mem ~inj:inj_int tbl "baaz"));
  assert_bool "not mem foo str" (not (mem ~inj:inj_str tbl "foo"));
  assert_bool "not mem bar str" (not (mem ~inj:inj_str tbl "bar"));
  assert_bool "mem baaz str" (mem ~inj:inj_str tbl "baaz");
  true
  ;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let inj_str = create_inj () in
  let tbl = create 5 in
  set ~inj:inj_int tbl "foo" 1;
  set ~inj:inj_int tbl "bar" 2;
  set ~inj:inj_str tbl "baaz" "hello";
  let l = keys_iter tbl |> Iter.to_list in
  assert_equal [ "baaz"; "bar"; "foo" ] (List.sort compare l);
  true
  ;;

  t @@ fun () ->
  let inj_int = create_inj () in
  let inj_str = create_inj () in
  let tbl = create 5 in
  set ~inj:inj_int tbl "foo" 1;
  set ~inj:inj_int tbl "bar" 2;
  set ~inj:inj_str tbl "baaz" "hello";
  set ~inj:inj_str tbl "str" "rts";
  let l_int = bindings_of ~inj:inj_int tbl |> Iter.to_list in
  assert_equal [ "bar", 2; "foo", 1 ] (List.sort compare l_int);
  let l_str = bindings_of ~inj:inj_str tbl |> Iter.to_list in
  assert_equal [ "baaz", "hello"; "str", "rts" ] (List.sort compare l_str);
  true
end

module Multiset = struct
  t @@ fun () ->
  let module S = CCMultiSet.Make (String) in
  S.count (S.add_mult S.empty "a" 5) "a" = 5
  ;;

  t @@ fun () ->
  let module S = CCMultiSet.Make (String) in
  S.count (S.remove_mult (S.add_mult S.empty "a" 5) "a" 3) "a" = 2
  ;;

  t @@ fun () ->
  let module S = CCMultiSet.Make (String) in
  S.count (S.remove_mult (S.add_mult S.empty "a" 4) "a" 6) "a" = 0
end

module PersistentArray = struct
  open CCPersistentArray;;

  t @@ fun () ->
  of_list
    [
      of_list [ 1 ];
      of_list [];
      of_list [ 2; 3; 4 ];
      of_list [ 5 ];
      of_list [ 6; 7 ];
    ]
  |> flatten |> to_list = [ 1; 2; 3; 4; 5; 6; 7 ]
  ;;

  t @@ fun () ->
  of_list [ of_list []; of_list []; of_list [] ] |> flatten |> length = 0
  ;;

  t @@ fun () -> of_list [] |> flatten |> length = 0;;
  q Q.(list int) (fun l -> of_list l |> to_gen |> of_gen |> to_list = l)
end