test-bigarrays: add case for float16

Author: igarnier

tests/test-bigarrays/dune

@@ -1,3 +1,9 @@
+(rule
+  (deps (:source test_bigarrays.cppo.ml))
+  (target test_bigarrays.ml)
+  (action (with-stdout-to %{target}
+            (run %{project_root}/src/mucppo/mucppo.exe %{source}))))
+
 (test
  (name test_bigarrays)
  (package ctypes-foreign)

tests/test-bigarrays/test_bigarrays.ml renamed to tests/test-bigarrays/test_bigarrays.cppo.ml

@@ -113,10 +113,10 @@ let test_bigarray_of_ctypes_array _ =
       [20L; 40L; 60L; 80L; 100L]];
      [[100L; 200L; 300L; 400L; 500L];
       [200L; 400L; 600L; 800L; 1000L]]] in
-     
+
   let b5 = BA.reshape
     (bigarray_of_array genarray BA.int64
-       (Array.from_ptr (castp int64_t (Array.start a5)) 30)) 
+       (Array.from_ptr (castp int64_t (Array.start a5)) 30))
     [| 3; 2; 5 |] in
   let () = begin
     assert_equal (Array.length a5) (BA.Genarray.nth_dim b5 0);
@@ -139,7 +139,7 @@ let test_bigarray_of_ctypes_array _ =
       [20.; 40.; 60.; 80.]];
      [[100.; 200.; 300.; 400.];
       [200.; 400.; 600.; 800.]]] in
-     
+
   let b6 = bigarray_of_array array3 BA.float64 a6 in
   let () = begin
     assert_equal (Array.length a6) (BA.Array3.dim1 b6);
@@ -171,7 +171,7 @@ let test_ctypes_array_of_bigarray _ =
 
     List.iteri (fun i -> BA.Genarray.set b1 [| i |])
       [ 6.; 5.; 4.; 3.; 2.; 1. ];
-    
+
     for i = 0 to b1_dim - 1 do
       assert_equal (BA.Genarray.get b1 [| i |]) a1.(i)
     done
@@ -187,7 +187,7 @@ let test_ctypes_array_of_bigarray _ =
     List.iteri (fun i ->
       fun v -> b2.{i} <- v)
       [ 2; 4; 6; 8; 10; 12; 14 ];
-    
+
     for i = 0 to b2_dim - 1 do
       assert_equal b2.{i} a2.(i)
     done
@@ -300,12 +300,12 @@ let test_ctypes_array_of_bigarray _ =
           [{re = 2.; im = 20.}; {re = 2e2; im = 0.0}];
           [{re = 3.; im = 30.}; {re = 3e2; im = 0.0}];
           [{re = 4.; im = 40.}; {re = 4e2; im = 0.0}]];
-         
+
          [[{re = 5.; im = 50.}; {re = 5e2; im = 0.1}];
           [{re = 6.; im = 60.}; {re = 6e2; im = 0.1}];
           [{re = 7.; im = 70.}; {re = 7e2; im = 0.1}];
           [{re = 8.; im = 80.}; {re = 8e2; im = 0.1}]];
-         
+
          [[{re = 9.; im = 90.}; {re = 9e2; im = 0.2}];
           [{re = 10.; im = 100.}; {re = 1e3; im = 0.2}];
           [{re = 11.; im = 110.}; {re = 1.1e3; im = 0.2}];
@@ -320,6 +320,23 @@ let test_ctypes_array_of_bigarray _ =
       done
     done
   end
+#if OCAML_VERSION >= (5, 2, 0)
+  ;
+  (* Copy of first test  case, with float16 instead of float32 *)
+  let b7_dim = 6 in
+  let b7 = BA.(Genarray.create float16 c_layout) [| b7_dim |] in
+  let a7 = array_of_bigarray genarray b7 in
+  begin
+    assert_equal (BA.Genarray.nth_dim b7 0) (Array.length a1);
+
+    List.iteri (fun i -> BA.Genarray.set b7 [| i |])
+      [ 6.; 5.; 4.; 3.; 2.; 1. ];
+
+    for i = 0 to b7_dim - 1 do
+      assert_equal (BA.Genarray.get b7 [| i |]) a7.(i)
+    done
+  end
+#endif
 
 
 (*