[OCaml] Accept context explicitly in Llvm_target functions

[oota-llvm.git] / test / Bindings / Ocaml / target.ml

diff --git a/test/Bindings/Ocaml/target.ml b/test/Bindings/Ocaml/target.ml
index f7d1cbf2859fa96b8eb8797854b33da21352ac90..017a886059c17b84d3ec80e65df10d2ba3320538 100644 (file)
--- a/test/Bindings/Ocaml/target.ml
+++ b/test/Bindings/Ocaml/target.ml
@@ -1,21 +1,31 @@
-(* RUN: %ocamlc -warn-error A llvm.cma llvm_target.cma %s -o %t 2> /dev/null
+(* RUN: rm -rf %t.builddir
+ * RUN: mkdir -p %t.builddir
+ * RUN: cp %s %t.builddir
+ * RUN: %ocamlopt -g -warn-error A llvm.cmxa llvm_target.cmxa %t.builddir/target.ml -o %t
+ * RUN: %t %t.bc
+ * XFAIL: vg_leak
  *)
 
-(* Note: It takes several seconds for ocamlc to link an executable with
+(* Note: It takes several seconds for ocamlopt to link an executable with
          libLLVMCore.a, so it's better to write a big test than a bunch of
          little ones. *)
 
 open Llvm
 open Llvm_target
 
+
 let context = global_context ()
 let i32_type = Llvm.i32_type context
 let i64_type = Llvm.i64_type context
 
 (* Tiny unit test framework - really just to help find which line is busted *)
-let suite name f =
-  prerr_endline (name ^ ":");
-  f ()
+let print_checkpoints = false
+
+let _ =
+  Printexc.record_backtrace true
+
+let assert_equal a b =
+  if a <> b then failwith "assert_equal"
 
 
 (*===-- Fixture -----------------------------------------------------------===*)
@@ -27,28 +37,28 @@ let m = create_module context filename
 (*===-- Target Data -------------------------------------------------------===*)
 
 let test_target_data () =
-  let td = TargetData.create (target_triple m) in
-  let sty = struct_type context [| i32_type; i64_type |] in
-  
-  ignore (TargetData.as_string td);
-  ignore (TargetData.invalidate_struct_layout td sty);
-  ignore (byte_order td);
-  ignore (pointer_size td);
-  ignore (intptr_type td);
-  ignore (size_in_bits td sty);
-  ignore (store_size td sty);
-  ignore (abi_size td sty);
-  ignore (stack_align td sty);
-  ignore (preferred_align td sty);
-  ignore (preferred_align_of_global td (declare_global sty "g" m));
-  ignore (element_at_offset td sty (Int64.of_int 1));
-  ignore (offset_of_element td sty 1);
+  let layout = "e-p:32:32:32-S32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-" ^
+               "f16:16:16-f32:32:32-f64:32:64-f128:128:128-v64:32:64-v128:32:128-" ^
+               "a0:0:64-n32" in
+  let td     = DataLayout.create layout in
+  let sty    = struct_type context [| i32_type; i64_type |] in
   
-  TargetData.dispose td
+  assert_equal (DataLayout.as_string td) layout;
+  assert_equal (byte_order td) Endian.Little;
+  assert_equal (pointer_size td) 4;
+  assert_equal (intptr_type td context) i32_type;
+  assert_equal (size_in_bits td sty) (Int64.of_int 96);
+  assert_equal (store_size td sty) (Int64.of_int 12);
+  assert_equal (abi_size td sty) (Int64.of_int 12);
+  assert_equal (stack_align td sty) 4;
+  assert_equal (preferred_align td sty) 8;
+  assert_equal (preferred_align_of_global td (declare_global sty "g" m)) 8;
+  assert_equal (element_at_offset td sty (Int64.of_int 1)) 0;
+  assert_equal (offset_of_element td sty 1) (Int64.of_int 4)
 
 
 (*===-- Driver ------------------------------------------------------------===*)
 
 let _ =
-  suite "target data" test_target_data;
+  test_target_data ();
   dispose_module m