def test_multi_env(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + multi_env_filename)
correct_env_kwargs_1 = {
'env_order': 1,
'env_method': 'pbe',
'fs_env_settings': {
'env_method': 'pbe'
}
}
correct_env_kwargs_2 = {
'env_order': 2,
'env_method': 'lda',
'fs_env_settings': {
'env_method': 'lda'
}
}
self.assertDictEqual(in_obj.supersystem_kwargs[0],
correct_env_kwargs_1)
self.assertDictEqual(in_obj.supersystem_kwargs[1],
correct_env_kwargs_2)
self.assertEqual(in_obj.env_subsystem_kwargs[0]['env_order'], 1)
self.assertEqual(in_obj.env_subsystem_kwargs[1]['env_order'], 1)
self.assertEqual(in_obj.env_subsystem_kwargs[2]['env_order'], 2)
self.assertEqual(in_obj.env_subsystem_kwargs[3]['env_order'], 2)
def test_hl_settings(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + hl_settings_filename)
correct_hl_kwargs = {
'hl_order': 1,
'hl_method': 'caspt2[10,10]',
'init_guess': 'submol',
'conv_tol': 1e-12,
'spin': 4,
'max_cycle': 100,
'damp': 10.,
'level_shift_factor': 12.1,
'use_ext': 'openmolcas',
'unrestricted': True,
'compress_approx': True,
'density_fitting': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
'hl_dict': {
'loc_orbs': True,
'cas_init_guess': 'rhf',
'active_orbs': [3, 4, 5, 6, 7],
'avas': ['1d']
}
}
self.assertEqual(len(in_obj.hl_subsystem_kwargs), 4)
for i in range(len(in_obj.hl_subsystem_kwargs)):
n = in_obj.hl_subsystem_kwargs[i]
if i == 0:
self.assertDictEqual(n, correct_hl_kwargs)
else:
self.assertEqual(n, None)
def test_subsys_hl_settings(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + subsys_hl_set_filename)
correct_hl_kwargs_1 = {
'hl_order': 1,
'hl_method': 'caspt2[10,10]',
'init_guess': 'minao',
'spin': 4,
'conv_tol': 1e-3,
'max_cycle': 4,
'damp': 0.001,
'level_shift_factor': 0.002,
'use_ext': 'molpro',
'unrestricted': True,
'compress_approx': True,
'density_fitting': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
'hl_dict': {
'cas_init_guess': 'ci',
'active_orbs': [5, 6, 7, 8, 9],
'avas': ['3d']
}
}
self.assertDictEqual(in_obj.hl_subsystem_kwargs[0],
correct_hl_kwargs_1)
def test_excited_kwargs(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + excited_filename)
correct_env_kwargs = {'env_order': 1, 'env_method': 'pbe'}
correct_hl_kwargs = {
'hl_order': 1,
'hl_method': 'ccsd',
'excited': True,
'hl_excited_dict': {
'nroots': 4
}
}
correct_supersystem_kwargs = {
'env_order': 1,
'env_method': 'pbe',
'excited_settings': {
'conv_tol': 1e-9,
'nroots': 4
},
'embed_settings': {
'excited_settings': {
'conv_tol': 1e-9,
'nroots': 4
},
'excited_relax': True,
'mu': 1000000.0
},
'fs_env_settings': {
'env_method': 'pbe',
'excited': True
}
}
self.assertEqual(len(in_obj.env_subsystem_kwargs), 4)
self.assertEqual(len(in_obj.hl_subsystem_kwargs), 4)
self.assertEqual(len(in_obj.supersystem_kwargs), 1)
for n in in_obj.env_subsystem_kwargs:
self.assertDictEqual(n, correct_env_kwargs)
for i, n in enumerate(in_obj.hl_subsystem_kwargs):
if i == 0:
self.assertDictEqual(n, correct_hl_kwargs)
else:
self.assertIsNone(n)
for n in in_obj.supersystem_kwargs:
self.assertDictEqual(n, correct_supersystem_kwargs)
def test_def_ecp(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + specify_ecp_filename)
self.maxDiff = None
correct_mol1 = gto.M()
correct_mol1.atom = '''
He 0.0000 0.0000 0.0000'''
correct_mol1.basis = '3-21g'
correct_mol1.ecp = 'lanl2tz'
correct_mol1.build()
correct_mol2 = gto.M()
correct_mol2.atom = '''
Fe 2.0000 0.0000 0.0000'''
correct_mol2.basis = '3-21g'
correct_mol2.ecp = 'lanl2dz'
correct_mol2.build()
correct_mol3 = gto.M()
correct_mol3.atom = '''
Fe-1 2.0000 2.0000 0.0000'''
correct_mol3.basis = '3-21g'
correct_mol3.ecp = 'lanl08'
correct_mol3.build()
correct_mol4 = gto.M()
correct_mol4.atom = '''
Fe-1 2.0000 2.0000 2.0000'''
correct_mol4.basis = '3-21g'
correct_mol4.ecp = 'lanl08'
correct_mol4.build()
corr_mol_list = [
correct_mol1, correct_mol2, correct_mol3, correct_mol4
]
self.assertEqual(len(in_obj.subsys_mols), 4)
for i in range(len(in_obj.subsys_mols)):
test = in_obj.subsys_mols[i]
corr = corr_mol_list[i]
self.assertListEqual(test._atom, corr._atom)
for k in corr._ecp.keys():
self.assertListEqual(test._ecp[k], corr._ecp[k])
def test_environ_settings(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + env_settings_filename)
fs_dict = {
'env_method': 'lda',
'init_guess': 'submol',
'conv_tol': 1e-6,
'max_cycle': 100,
'damp': 0.2,
'level_shift_factor': 1.4,
'diis_num': 1,
'grid_level': 4,
'small_rho_cutoff': 1e-1,
'unrestricted': True,
'dynamic_level_shift': True,
'density_fitting': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
}
ft_dict = {
'max_cycle': 12,
'subcycles': 2,
'conv_tol': 1e-4,
'compare_density': True,
'damp': 0.5,
'diis_num': 3,
'set_fermi': 0.35,
'update_fock': 3,
'update_proj': 2,
'init_guess': 'chk',
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
'proj_oper': 'huzfermi',
}
correct_supersystem_kwargs = {'env_order': 1, 'env_method': 'lda'}
correct_supersystem_kwargs['fs_env_settings'] = fs_dict
correct_supersystem_kwargs['embed_settings'] = ft_dict
self.assertEqual(len(in_obj.supersystem_kwargs), 1)
for n in in_obj.supersystem_kwargs:
self.assertDictEqual(n, correct_supersystem_kwargs)
def test_subsys_env_settings(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + subsys_env_set_filename)
correct_env_kwargs_1 = {
'env_order': 1,
'env_method': 'lda',
'smearsigma': 0.2,
'init_guess': 'chk',
'conv_tol': 1e-20,
'damp': 12.2,
'level_shift_factor': 2.3,
'subcycles': 10,
'diis_num': 6,
'unrestricted': True,
'density_fitting': True,
'freeze': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
}
self.assertDictEqual(in_obj.env_subsystem_kwargs[1],
correct_env_kwargs_1)
def test_def_basis(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + specify_basis_filename)
self.maxDiff = None
correct_mol1 = gto.M()
correct_mol1.atom = '''
He 0.0000 0.0000 0.0000'''
correct_mol1.basis = '3-21g'
correct_mol1.build()
correct_mol2 = gto.M()
correct_mol2.atom = '''
C 2.0000 0.0000 0.0000'''
correct_mol2.basis = 'sto-3g'
correct_mol2.build()
correct_mol3 = gto.M()
correct_mol3.atom = '''
C-1 2.0000 2.0000 0.0000'''
correct_mol3.basis = 'cc-pVDZ'
correct_mol3.build()
correct_mol4 = gto.M()
correct_mol4.atom = '''
C-1 2.0000 2.0000 2.0000'''
correct_mol4.basis = 'cc-pVDZ'
correct_mol4.build()
corr_mol_list = [
correct_mol1, correct_mol2, correct_mol3, correct_mol4
]
self.assertEqual(len(in_obj.subsys_mols), 4)
for i in range(len(in_obj.subsys_mols)):
test = in_obj.subsys_mols[i]
corr = corr_mol_list[i]
self.assertListEqual(test._atom, corr._atom)
for k in test._atom:
self.assertListEqual(test._basis[k[0]], corr._basis[k[0]])
def test_spin_charge(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + def_filename)
correct_mol1 = gto.M()
correct_mol1.atom = '''
He 0.0000 0.0000 0.0000'''
correct_mol1.basis = '3-21g'
correct_mol1.build()
correct_mol2 = gto.M()
correct_mol2.atom = '''
C 2.0000 0.0000 0.0000'''
correct_mol2.basis = '3-21g'
correct_mol2.build()
correct_mol3 = gto.M()
correct_mol3.atom = '''
C 2.0000 2.0000 0.0000'''
correct_mol3.basis = '3-21g'
correct_mol3.build()
correct_mol4 = gto.M()
correct_mol4.atom = '''
C 2.0000 2.0000 2.0000'''
correct_mol4.basis = '3-21g'
correct_mol4.build()
corr_mol_list = [
correct_mol1, correct_mol2, correct_mol3, correct_mol4
]
self.assertEqual(len(in_obj.subsys_mols), 4)
for i in range(len(in_obj.subsys_mols)):
test = in_obj.subsys_mols[i]
corr = corr_mol_list[i]
self.assertListEqual(test._atom, corr._atom)
self.assertDictEqual(test._basis, corr._basis)
def test_default_inp(self):
path = os.getcwd() + temp_inp_dir #Maybe a better way
in_obj = inp_reader.InpReader(path + def_filename)
correct_env_kwargs = {'env_order': 1, 'env_method': 'pbe'}
correct_hl_kwargs = {'hl_order': 1, 'hl_method': 'rhf'}
correct_supersystem_kwargs = {
'env_order': 1,
'env_method': 'pbe',
'fs_env_settings': {
'env_method': 'pbe'
}
}
self.assertEqual(len(in_obj.env_subsystem_kwargs), 4)
self.assertEqual(len(in_obj.hl_subsystem_kwargs), 4)
self.assertEqual(len(in_obj.supersystem_kwargs), 1)
for n in in_obj.env_subsystem_kwargs:
self.assertDictEqual(n, correct_env_kwargs)
for i, n in enumerate(in_obj.hl_subsystem_kwargs):
if i == 0:
self.assertDictEqual(n, correct_hl_kwargs)
else:
self.assertIsNone(n)
for n in in_obj.supersystem_kwargs:
self.assertDictEqual(n, correct_supersystem_kwargs)
def test_explicit_inp(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + explicit_filename)
correct_env_kwargs_1 = {
'env_order': 3,
'env_method': 'pbe',
}
correct_env_kwargs_2 = {
'env_order': 3,
'env_method': 'pbe',
}
correct_env_kwargs_3 = {
'env_order': 2,
'env_method': 'pbe',
'smearsigma': 0.1,
'init_guess': 'atom',
'conv_tol': 1e-3,
'damp': 0.5,
'level_shift_factor': 1.,
'subcycles': 10,
'diis_num': 1,
'density_fitting': True,
'freeze': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True,
}
correct_env_kwargs_4 = {
'env_order': 3,
'env_method': 'pbe',
'unrestricted': True,
}
correct_env_kwargs_5 = {
'env_order': 1,
'env_method': 'lda',
'subcycles': 2,
'unrestricted': True
}
env_list = [
correct_env_kwargs_1, correct_env_kwargs_2, correct_env_kwargs_3,
correct_env_kwargs_4, correct_env_kwargs_5
]
correct_hl_kwargs_1 = {
'hl_order': 1,
'hl_method': 'cas[2,2]',
'init_guess': 'atom',
'spin': 2,
'conv_tol': 1e-3,
'max_cycle': 100,
'damp': 0.8,
'level_shift_factor': 2.,
'use_ext': 'openmolcas',
'unrestricted': True,
'compress_approx': True,
'density_fitting': True,
'hl_dict': {
'loc_orbs': True,
'cas_init_guess': 'rhf',
'active_orbs': [1, 2, 3],
'avas': ['1d']
}
}
correct_hl_kwargs_2 = {
'hl_order': 2,
'hl_method': 'rhf',
'save_spin_density': True
}
hl_list = [correct_hl_kwargs_1, correct_hl_kwargs_2, None, None, None]
correct_sup_kwargs_1 = {
'env_order': 1,
'env_method': 'lda',
'fs_env_settings': {
'env_method': 'lda',
'init_guess': 'supmol',
'conv_tol': 1e-1,
'max_cycle': 1,
'damp': 0.0001,
'level_shift_factor': 20.0,
'diis_num': 4,
'grid_level': 7,
'small_rho_cutoff': 1.,
'unrestricted': True,
'density_fitting': True,
'save_orbs': True,
'save_density': True,
'save_spin_density': True
},
'embed_settings': {
'compare_density': True,
'max_cycle': 100,
'subcycles': 2,
'conv_tol': 1e-4,
'damp': 0.1,
'diis_num': 1,
'set_fermi': 3.,
'update_fock': 1,
'init_guess': 'submol',
'unrestricted': True,
'save_orbs': True,
'save_density': True,
'proj_oper': 1e-5,
},
}
correct_sup_kwargs_2 = {
'env_order': 2,
'env_method': 'pbe',
'fs_env_settings': {
'env_method': 'pbe',
}
}
correct_sup_kwargs_3 = {
'env_order': 3,
'env_method': 'pbe',
'fs_env_settings': {
'env_method': 'pbe'
}
}
sup_list = [
correct_sup_kwargs_1, correct_sup_kwargs_2, correct_sup_kwargs_3
]
self.assertEqual(len(in_obj.env_subsystem_kwargs), 5)
self.assertEqual(len(in_obj.hl_subsystem_kwargs), 5)
self.assertEqual(len(in_obj.supersystem_kwargs), 3)
for i in range(len(in_obj.env_subsystem_kwargs)):
test = in_obj.env_subsystem_kwargs[i]
self.assertDictEqual(test, env_list[i])
print(f"ENV SUBSYSTEM {i} GOOD")
for i in range(len(in_obj.hl_subsystem_kwargs)):
test = in_obj.hl_subsystem_kwargs[i]
self.assertEqual(test, hl_list[i])
print(f"HL SUBSYSTEM {i} GOOD")
for i in range(len(in_obj.supersystem_kwargs)):
test = in_obj.supersystem_kwargs[i]
self.assertDictEqual(test, sup_list[i])
print(f"SUPERSYSTEM {i} GOOD")
def test_explicit(self):
path = os.getcwd() + temp_inp_dir
in_obj = inp_reader.InpReader(path + explicit_filename)
correct_mol1 = gto.M()
correct_mol1.atom = '''
He 0.0000 0.0000 0.0000'''
correct_mol1.basis = '3-21g'
correct_mol1.build()
correct_mol2 = gto.M()
correct_mol2.atom = '''
He-1 2.0000 0.0000 0.0000
He 2.0000 2.0000 0.0000
He 2.0000 2.0000 2.0000'''
correct_mol2.charge = 2
correct_mol2.basis = {'He': 'sto-3g', 'He-1': '6-31g'}
correct_mol2.build()
correct_mol3 = gto.M()
correct_mol3.atom = '''
Fe 4.0000 0.0000 0.0000
Fe 4.0000 2.0000 0.0000
Fe 4.0000 2.0000 2.0000
ghost:Fe 4.0000 4.0000 2.0000'''
correct_mol3.basis = 'cc-pVDZ'
correct_mol3.ecp = 'lanl2dz'
correct_mol3.unit = 'bohr'
correct_mol3.build()
correct_mol4 = gto.M()
correct_mol4.atom = '''
H-1 8.0000 0.0000 0.0000
H 8.0000 2.0000 0.0000
H 8.0000 2.0000 2.0000'''
correct_mol4.basis = {'H': 'aug-cc-pVDZ', 'H-1': 'cc-pVTZ'}
correct_mol4.spin = 3
correct_mol4.build()
correct_mol5 = gto.M()
correct_mol5.atom = '''
H-1 10.0000 0.0000 0.0000
H 10.0000 2.0000 0.0000
H 10.0000 2.0000 2.0000'''
correct_mol5.basis = {'H': 'aug-cc-pVDZ', 'H-1': 'cc-pVTZ'}
correct_mol5.spin = 3
correct_mol5.build()
corr_mol_list = [
correct_mol1, correct_mol2, correct_mol3, correct_mol4,
correct_mol5
]
self.assertEqual(len(in_obj.subsys_mols), 5)
for i in range(len(in_obj.subsys_mols)):
test = in_obj.subsys_mols[i]
corr = corr_mol_list[i]
self.assertListEqual(test._atom, corr._atom)
self.assertEqual(test.charge, corr.charge)
self.assertEqual(test.spin, corr.spin)
for k in test._atom:
self.assertListEqual(test._basis[k[0]], corr._basis[k[0]])
for k in test._ecp.keys():
self.assertListEqual(test._ecp[k], corr._ecp[k])
def setUp(self):
self.prepare_input(he_chain_input)
self.in_obj = inp_reader.InpReader(self.path + self.filename)
def setUp(self):
self.prepare_input(supersystem_basis_input)
self.in_obj = inp_reader.InpReader(self.path + self.filename)
