mirror of
https://gitlab.com/scemama/QCaml.git
synced 2024-07-18 08:53:27 +02:00
126 lines
3.2 KiB
OCaml
126 lines
3.2 KiB
OCaml
|
open Alcotest
|
||
|
open Qcaml_common
|
||
|
open Qcaml_gaussian_integrals
|
||
|
open Qcaml_gaussian_basis
|
||
|
open Qcaml_ao_basis
|
||
|
open Qcaml_particles
|
||
|
open Qcaml_operators
|
||
|
open Qcaml_linear_algebra
|
||
|
open Ao_basis_gaussian
|
||
|
|
||
|
let wd = Qcaml.root ^ Filename.dir_sep ^ "test"
|
||
|
|
||
|
let make_tests name t =
|
||
|
|
||
|
let check_matrix title a r =
|
||
|
let ax =
|
||
|
Matrix.as_vec_inplace a
|
||
|
|> Vector.to_bigarray_inplace
|
||
|
in
|
||
|
Matrix.as_vec_inplace r
|
||
|
|> Vector.iteri (fun i x ->
|
||
|
let message =
|
||
|
Printf.sprintf "%s line %d" title i
|
||
|
in
|
||
|
check (float 1.e-10) message ax.{i} x
|
||
|
)
|
||
|
in
|
||
|
|
||
|
let check_eri a r =
|
||
|
let f { Eri.i_r1 ; j_r2 ; k_r1 ; l_r2 ; value } =
|
||
|
(i_r1, (j_r2, (k_r1, (l_r2, value))))
|
||
|
in
|
||
|
let a = Eri.to_list a |> List.rev_map f in
|
||
|
let r = Eri.to_list r |> List.rev_map f in
|
||
|
check (list (pair int (pair int (pair int (pair int (float 1.e-10)))))) "ERI" a r
|
||
|
in
|
||
|
|
||
|
let check_eri_lr a r =
|
||
|
let f { Eri_long_range.i_r1 ; j_r2 ; k_r1 ; l_r2 ; value } =
|
||
|
(i_r1, (j_r2, (k_r1, (l_r2, value))))
|
||
|
in
|
||
|
let a = Eri_long_range.to_list a |> List.rev_map f in
|
||
|
let r = Eri_long_range.to_list r |> List.rev_map f in
|
||
|
check (list (pair int (pair int (pair int (pair int (float 1.e-10)))))) "ERI_lr" a r
|
||
|
in
|
||
|
|
||
|
let test_overlap () =
|
||
|
let reference =
|
||
|
Matrix.sym_matrix_of_file (wd ^ Filename.dir_sep ^ name ^ "_overlap.ref")
|
||
|
in
|
||
|
let overlap =
|
||
|
Lazy.force t.overlap |> Overlap.matrix
|
||
|
in
|
||
|
check_matrix "Overlap" overlap reference
|
||
|
in
|
||
|
|
||
|
let test_eN_ints () =
|
||
|
let reference =
|
||
|
Matrix.sym_matrix_of_file (wd ^ Filename.dir_sep ^ name ^ "_nuc.ref")
|
||
|
in
|
||
|
let eN_ints =
|
||
|
Lazy.force t.eN_ints |> Electron_nucleus.matrix
|
||
|
in
|
||
|
check_matrix "eN_ints" eN_ints reference
|
||
|
in
|
||
|
|
||
|
let test_kin_ints () =
|
||
|
let reference =
|
||
|
Matrix.sym_matrix_of_file (wd ^ Filename.dir_sep ^ name ^ "_kin.ref")
|
||
|
in
|
||
|
let kin_ints =
|
||
|
Lazy.force t.kin_ints |> Kinetic.matrix
|
||
|
in
|
||
|
check_matrix "kin_ints" kin_ints reference
|
||
|
in
|
||
|
|
||
|
let test_ee_ints () =
|
||
|
let reference =
|
||
|
Eri.of_file (wd ^ Filename.dir_sep ^ name ^ "_eri.ref") ~sparsity:`Dense ~size:(Basis.size t.basis)
|
||
|
in
|
||
|
let ee_ints =
|
||
|
Lazy.force t.ee_ints
|
||
|
in
|
||
|
check_eri ee_ints reference
|
||
|
;
|
||
|
in
|
||
|
|
||
|
let test_ee_lr_ints () =
|
||
|
let reference =
|
||
|
Eri_long_range.of_file (wd ^ Filename.dir_sep ^ name ^ "_eri_lr.ref") ~sparsity:`Dense
|
||
|
~size:(Basis.size t.basis)
|
||
|
in
|
||
|
let ee_lr_ints =
|
||
|
Lazy.force t.ee_lr_ints
|
||
|
in
|
||
|
check_eri_lr ee_lr_ints reference
|
||
|
in
|
||
|
|
||
|
[
|
||
|
"Overlap", `Quick, test_overlap;
|
||
|
"eN_ints", `Quick, test_eN_ints;
|
||
|
"kin_ints", `Quick, test_kin_ints;
|
||
|
"ee_ints", `Quick, test_ee_ints;
|
||
|
"ee_lr_ints", `Quick, test_ee_lr_ints;
|
||
|
]
|
||
|
|
||
|
let tests =
|
||
|
List.concat [
|
||
|
let nuclei =
|
||
|
wd ^ Filename.dir_sep ^ "water.xyz"
|
||
|
|> Nuclei.of_xyz_file
|
||
|
in
|
||
|
let basis =
|
||
|
wd ^ Filename.dir_sep ^ "cc-pvdz"
|
||
|
|> Basis.of_nuclei_and_basis_filename ~nuclei
|
||
|
in
|
||
|
let rs = 0.5 in
|
||
|
let operators = [ Operator.of_range_separation rs ] in
|
||
|
let ao_basis =
|
||
|
Ao_basis_gaussian.make ~basis ~operators nuclei
|
||
|
in
|
||
|
make_tests "water" ao_basis ;
|
||
|
|
||
|
]
|
||
|
|