open OUnit2
module N = DataType.Num

let test_num_add n1 n2 result ctx = begin
  assert_equal
    ~cmp:(=)
    result
    (N.to_int @@ N.add n1 n2)
end

let test_num_mult n1 n2 result ctx = begin
  assert_equal
    ~cmp:(=)
    result
    (N.to_int @@ N.mult n1 n2)
end

let test_num_sub n1 n2 result ctx = begin
  assert_equal
    ~cmp:(=)
    result
    (N.to_int @@ N.sub n1 n2)
end

let test_floor n1 expected ctx = begin

  let result = N.to_int @@ N.floor n1 in

  assert_equal
    ~msg:(Printf.sprintf "Expected %d but got %d" expected result)
    ~cmp:(=)
    expected
    result

end



let n1 = N.of_int 1
let n2 = N.of_int 2

let num_tests = "num_test">::: [

    "test_add"      >:: test_num_add n1     n1    2;
    "test_mult"     >:: test_num_mult n2    n1    2;
    "test_sub"      >:: test_num_sub n1     n1    0;
    "test_floor1"   >:: test_floor (N.of_float   1.2)    1;
    "test_floor2"   >:: test_floor (N.of_float (-1.2)) (-2);
 ]