From ee4ceb06c1efa8c77093cde8b31decaedb32c4bc Mon Sep 17 00:00:00 2001
From: Nick Mayhall <nmayhall@vt.edu>
Date: Fri, 28 Oct 2022 11:18:05 -0400
Subject: [PATCH] added test file

---
 test/test_gd.jl | 117 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 test/test_gd.jl

diff --git a/test/test_gd.jl b/test/test_gd.jl
new file mode 100644
index 0000000..a1955b8
--- /dev/null
+++ b/test/test_gd.jl
@@ -0,0 +1,117 @@
+using QCBase  
+using RDM  
+using ClusterMeanField  
+using LinearAlgebra
+using Printf
+using Test
+using NPZ
+using InCoreIntegrals
+using ActiveSpaceSolvers
+
+    
+@testset "gd open shell" begin
+    atoms = []
+    push!(atoms,Atom(1,"H",[0,0,0]))
+    push!(atoms,Atom(2,"H",[1,0,0]))
+    push!(atoms,Atom(3,"H",[2,0,0]))
+    push!(atoms,Atom(4,"H",[3,1,0]))
+    push!(atoms,Atom(5,"H",[4,1,0]))
+    push!(atoms,Atom(6,"H",[5,1,0]))
+    #basis = "6-31g"
+    basis = "sto-3g"
+
+    mol     = Molecule(0,1,atoms,basis)
+    mf = ClusterMeanField.pyscf_do_scf(mol)
+    nbas = size(mf.mo_coeff)[1]
+    ints = ClusterMeanField.pyscf_build_ints(mol,mf.mo_coeff, zeros(nbas,nbas));
+    e_fci, d1a_fci, d1b_fci,d2_fci = ClusterMeanField.pyscf_fci(ints,3,3)
+    @printf(" HF  Energy: %12.8f\n", mf.e_tot)
+    @printf(" FCI Energy: %12.8f\n", e_fci+ints.h0)
+
+    ClusterMeanField.pyscf_write_molden(mol,mf.mo_coeff,filename="scf.molden")
+
+    C = mf.mo_coeff
+
+    d1_fci = RDM1(d1a_fci,d1b_fci)
+    
+    d1_fci = ssRDM1(d1a_fci+d1b_fci)
+    d2_fci = ssRDM2(d2_fci)
+
+    rdm_mf = mf.mo_coeff'*mf.get_ovlp()*mf.make_rdm1()*mf.get_ovlp()*mf.mo_coeff / 2
+    rdm1 = RDM1(rdm_mf, rdm_mf)
+    
+    @printf(" Should be E(HF):  %12.8f\n", compute_energy(ints, rdm1, RDM2(rdm1)))
+    @printf(" Should be E(FCI): %12.8f\n", compute_energy(ints, d1_fci, d2_fci))
+    
+    Cl = ClusterMeanField.localize(mf.mo_coeff,"lowdin",mf)
+    ClusterMeanField.pyscf_write_molden(mol,Cl,filename="lowdin.molden")
+    S = ClusterMeanField.get_ovlp(mf)
+    U =  C' * S * Cl
+    println(" Build Integrals")
+    flush(stdout)
+    ints = orbital_rotation(ints,U)
+    rdm1 = orbital_rotation(rdm1,U)
+    d1_fci = orbital_rotation(d1_fci,U)
+    d2_fci = orbital_rotation(d2_fci,U)
+    println(" done.")
+    flush(stdout)
+
+
+    @printf(" Should be E(HF):  %12.8f\n", compute_energy(ints, rdm1, RDM2(rdm1)))
+    @printf(" Should be E(FCI): %12.8f\n", compute_energy(ints, d1_fci, d2_fci))
+    
+    e_fci = -3.155304800477
+    e_scf = -3.09169726403968
+   
+    sol = solve(ints, FCIAnsatz(6,3,3), SolverSettings(nroots=1, tol=1e-8))
+    display(sol)
+    d1a, d1b, d2aa, d2bb, d2ab = compute_1rdm_2rdm(sol) 
+    
+    d1_fci = RDM1(d1a, d1b)
+    d2_fci = RDM2(d2aa, d2ab, d2bb)
+
+    clusters    = [(1:3),(4:6)]
+    init_fspace = [(2,1),(1,2)]
+
+    clusters = [MOCluster(i,collect(clusters[i])) for i = 1:length(clusters)]
+    display(clusters)
+
+    n = n_orb(ints)
+    
+    e_cmf, U, d1 = ClusterMeanField.cmf_oo(ints, clusters, init_fspace, rdm1, 
+                              verbose=0, gconv=1e-10, sequential=true)
+    @test isapprox(e_cmf, -3.104850893117, atol=1e-12)
+   
+    #ints = orbital_rotation(ints, U)
+
+    e_cmf, Ugd, d1gd = ClusterMeanField.cmf_oo_diis(ints, clusters, init_fspace, rdm1, 
+                              tol_oo=1e-10, 
+                              tol_d1=1e-11, 
+                              tol_ci=1e-12, 
+                              alpha=.1, 
+                              sequential=true, 
+                              max_ss_size=8)
+    
+    display(U'*Ugd)
+    display(d1.a - d1gd.a)
+
+    @test isapprox(e_cmf, -3.104850893117, atol=1e-12)
+   
+    e_cmf, Ugd, d1gd = ClusterMeanField.cmf_oo_gd(ints, clusters, init_fspace, rdm1, 
+                              tol_oo=1e-10, 
+                              tol_d1=1e-12, 
+                              tol_ci=1e-14, 
+                              alpha=.2, 
+                              sequential=true)
+    
+    display(U'*Ugd)
+    display(d1.a - d1gd.a)
+
+    @test isapprox(e_cmf, -3.104850893117, atol=1e-12)
+    
+    return
+    e_cmf, Ugd, d1gd = ClusterMeanField.cmf_oo_gd(ints, clusters, init_fspace, rdm1, 
+                              verbose=0, conv_oo=1e-10, conv_ci=1e-11, sequential=true, alpha=.2)
+
+end
+