open OUnit

let test_cardinal () =
  let bv1 = BV.create ~size:87 true in
  assert_equal ~printer:string_of_int 87 (BV.cardinal bv1);
  ()

let test_get () =
  let bv = BV.create ~size:99 false in
  assert_bool "32 must be false" (not (BV.get bv 32));
  assert_bool "88 must be false" (not (BV.get bv 88));
  assert_bool "5 must be false" (not (BV.get bv 5));
  BV.set bv 32;
  BV.set bv 88;
  BV.set bv 5;
  assert_bool "32 must be true" (BV.get bv 32);
  assert_bool "88 must be true" (BV.get bv 88);
  assert_bool "5 must be true" (BV.get bv 5);
  assert_bool "33 must be false" (not (BV.get bv 33));
  assert_bool "44 must be false" (not (BV.get bv 44));
  assert_bool "1 must be false" (not (BV.get bv 1));
  ()

let test_list () =
  let bv = BV.of_list [1; 5; 156; 0; 222] in
  assert_equal ~printer:string_of_int 5 (BV.cardinal bv);
  BV.set bv 201;
  assert_equal ~printer:string_of_int 6 (BV.cardinal bv);
  let l = BV.to_list bv in
  let l = List.sort compare l in
  assert_equal [0;1;5;156;201;222] l;
  ()

let test_clear () =
  let bv = BV.of_list [1; 5; 200] in
  assert_equal ~printer:string_of_int 3 (BV.cardinal bv);
  BV.clear bv;
  assert_equal ~printer:string_of_int 0 (BV.cardinal bv);
  assert_bool "must be empty" (BV.is_empty bv);
  ()

let test_union () =
  let bv1 = BV.of_list [1;2;3;4] in
  let bv2 = BV.of_list [4;200;3] in
  let bv = BV.union bv1 bv2 in
  let l = List.sort compare (BV.to_list bv) in
  assert_equal [1;2;3;4;200] l;
  ()

let test_inter () =
  let bv1 = BV.of_list [1;2;3;4] in
  let bv2 = BV.of_list [4;200;3] in
  BV.inter_into ~into:bv1 bv2;
  let l = List.sort compare (BV.to_list bv1) in
  assert_equal [3;4] l;
  ()

let test_select () =
  let bv = BV.of_list [1;2;5;400] in
  let arr = [|"a"; "b"; "c"; "d"; "e"; "f"|] in
  let l = List.sort compare (BV.select bv arr) in
  assert_equal [("b",1); ("c",2); ("f",5)] l;
  ()

let suite = "test_bv" >:::
  [ "test_cardinal" >:: test_cardinal
  ; "test_get" >:: test_get
  ; "test_list" >:: test_list
  ; "test_clear" >:: test_clear
  ; "test_union" >:: test_union
  ; "test_inter" >:: test_inter
  ; "test_select" >:: test_select
  ]