1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# fmt: off
import pytest
from ase.build import molecule
calc = pytest.mark.calculator
@calc('cp2k', auto_write=True, uks=True)
@calc('gamess_us')
@calc('gaussian')
def test_h2dft(factory):
name = factory.name
calc = factory.calc(label=name, xc='LDA')
h2 = molecule('H2', calculator=calc)
h2.center(vacuum=2.0)
e2 = h2.get_potential_energy()
calc.set(xc='PBE')
e2pbe = h2.get_potential_energy()
h1 = h2.copy()
del h1[1]
h1.set_initial_magnetic_moments([1])
h1.calc = calc
e1pbe = h1.get_potential_energy()
calc.set(xc='LDA')
e1 = h1.get_potential_energy()
try:
m1 = h1.get_magnetic_moment()
except NotImplementedError:
pass
else:
print(m1)
print(2 * e1 - e2)
print(2 * e1pbe - e2pbe)
print(e1, e2, e1pbe, e2pbe)
calc = factory.calc(restart=name)
print(calc.parameters, calc.results, calc.atoms)
assert not calc.calculation_required(h1, ['energy'])
h1 = calc.get_atoms()
print(h1.get_potential_energy())
label = 'dir/' + name + '-h1'
calc = factory.calc(label=label, atoms=h1, xc='LDA')
print(h1.get_potential_energy())
|
