mirror of
https://gitlab.com/scemama/QCaml.git
synced 2024-06-02 03:15:19 +02:00
60 lines
2.6 KiB
OCaml
60 lines
2.6 KiB
OCaml
open Linear_algebra
|
|
open Alcotest
|
|
open Lacaml.D
|
|
|
|
let test_all () =
|
|
let n = 100 in
|
|
let a1 = Array.init n (fun _ -> Random.float 10. -. 5.) in
|
|
let a2 = Array.init n (fun _ -> Random.float 10. -. 5.) in
|
|
let u1 = Vec.of_array a1 in
|
|
let u2 = Vec.of_array a2 in
|
|
let v1 = Vector.of_array a1 in
|
|
let v2 = Vector.of_array a2 in
|
|
|
|
let check_dot1 label f1 f2 =
|
|
check (float 1.e-14) label (dot u1 (f1 u2)) (Vector.dot v1 (f2 v2))
|
|
in
|
|
|
|
let check_dot2 label f1 f2 =
|
|
check (float 1.e-14) label (dot u1 (f1 u1 u2)) (Vector.dot v1 (f2 v1 v2))
|
|
in
|
|
|
|
check int "dim" (Array.length a1) (Vector.dim v1);
|
|
check (float 1.e-14) "dot" (dot u1 u2) (Vector.dot v1 v2);
|
|
check_dot2 "add" (fun x y -> Vec.add x y) Vector.add ;
|
|
check_dot2 "sub" (fun x y -> Vec.sub x y) Vector.sub ;
|
|
check_dot2 "mul" (fun x y -> Vec.mul x y) Vector.mul ;
|
|
check_dot2 "div" (fun x y -> Vec.div x y) Vector.div ;
|
|
check_dot1 "sqr" (fun x -> Vec.sqr x) Vector.sqr ;
|
|
check_dot1 "sin" (fun x -> Vec.sin x) Vector.sin ;
|
|
check_dot1 "cos" (fun x -> Vec.cos x) Vector.cos ;
|
|
check_dot1 "tan" (fun x -> Vec.tan x) Vector.tan ;
|
|
check_dot1 "abs" (fun x -> Vec.abs x) Vector.abs ;
|
|
check_dot1 "neg" (fun x -> Vec.neg x) Vector.neg ;
|
|
check_dot1 "asin" (fun x -> Vec.asin x) Vector.asin ;
|
|
check_dot1 "acos" (fun x -> Vec.acos x) Vector.acos ;
|
|
check_dot1 "atan" (fun x -> Vec.atan x) Vector.atan ;
|
|
check_dot1 "sqrt" (fun x -> Vec.sqrt x) Vector.sqrt ;
|
|
check_dot1 "reci" (fun x -> Vec.reci x) Vector.reci ;
|
|
check_dot1 "map" (fun x -> Vec.map (fun y -> y+. 3.) x) (Vector.map (fun y -> y+. 3.)) ;
|
|
check (float 1.e-14) "norm" (sqrt (dot u1 u1)) (Vector.norm v1);
|
|
check (float 1.e-14) "norm" (sqrt (dot u2 u2)) (Vector.norm v2);
|
|
check (float 1.e-14) "sum" (Vec.sum u1) (Vector.sum v1);
|
|
check (float 1.e-14) "sum" (Vec.sum u2) (Vector.sum v2);
|
|
check (float 1.e-14) "at" (u1.{n/2}) (Vector.at v1 (n/2));
|
|
check (float 1.e-14) "at" (u2.{n/2}) (Vector.at v2 (n/2));
|
|
check (bool) "of_list" true (v1 = Vector.of_list @@ Array.to_list a1);
|
|
check (bool) "of_list" true (v2 = Vector.of_list @@ Array.to_list a2);
|
|
check (bool) "to_list" true (Vector.to_list v1 = Array.to_list a1);
|
|
check (bool) "to_list" true (Vector.to_list v2 = Array.to_list a2);
|
|
check (bool) "to_array" true (Vector.to_array v1 = a1);
|
|
check (bool) "to_array" true (Vector.to_array v2 = a2);
|
|
check (bool) "make0" true (Vector.make0 n = Vector.make n 0.);
|
|
check (float 1.e-14) "fold" (Vector.sum v1) (Vector.fold (fun a x -> a +. x) 0. v1);
|
|
check (float 1.e-14) "fold" (Vector.sum v2) (Vector.fold (fun a x -> a +. x) 0. v2);
|
|
()
|
|
|
|
let tests = [
|
|
"string", `Quick, test_all;
|
|
]
|