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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
import pytest
import numpy as np
from ase.build import bulk, molecule
from ase.units import Hartree
calc = pytest.mark.calculator
required_quantities = {'eigenvalues',
'fermilevel',
'version',
'forces',
'energy',
'free_energy',
'stress',
'ibz_kpoints',
'kpoint_weights'}
def run(atoms):
atoms.get_forces()
print(sorted(atoms.calc.results))
for key, value in atoms.calc.results.items():
if isinstance(value, np.ndarray):
print(key, value.shape, value.dtype)
else:
print(key, value)
for name in required_quantities:
assert name in atoms.calc.results
return atoms.calc.results
@pytest.mark.calculator_lite
@calc('abinit')
@calc('abinit', v8_legacy_format=False)
def test_si(factory):
atoms = bulk('Si')
atoms.calc = factory.calc(nbands=4 * len(atoms), kpts=[4, 4, 4])
run(atoms)
@pytest.mark.calculator_lite
@pytest.mark.parametrize('pps', ['fhi', 'paw'])
@calc('abinit')
def test_au(factory, pps):
atoms = bulk('Au')
atoms.calc = factory.calc(
pps=pps,
nbands=10 * len(atoms),
tsmear=0.1,
occopt=3,
kpts=[2, 2, 2],
pawecutdg=6.0 * Hartree,
)
# Somewhat awkward to set pawecutdg also when we are not doing paw,
# but it's an error to pass None as pawecutdg.
run(atoms)
@pytest.fixture
def fe_atoms(abinit_factory):
atoms = bulk('Fe')
atoms.set_initial_magnetic_moments([1])
calc = abinit_factory.calc(nbands=8,
kpts=[2, 2, 2])
atoms.calc = calc
return atoms
def test_fe_fixed_magmom(fe_atoms):
fe_atoms.calc.set(spinmagntarget=2.3)
run(fe_atoms)
@pytest.mark.calculator_lite
def test_fe_any_magmom(fe_atoms):
fe_atoms.calc.set(occopt=7)
run(fe_atoms)
@calc('abinit')
def test_h2o(factory):
atoms = molecule('H2O', vacuum=2.5)
atoms.calc = factory.calc(nbands=8)
run(atoms)
@calc('abinit')
def test_o2(factory):
atoms = molecule('O2', vacuum=2.5)
atoms.calc = factory.calc(nbands=8, occopt=7)
run(atoms)
magmom = atoms.get_magnetic_moment()
assert magmom == pytest.approx(2, 1e-2)
print('magmom', magmom)
@pytest.mark.skip('expensive')
@calc('abinit')
def test_manykpts(factory):
atoms = bulk('Au') * (2, 2, 2)
atoms.rattle(stdev=0.01)
atoms.symbols[:2] = 'Cu'
atoms.calc = factory.calc(nbands=len(atoms) * 7, kpts=[8, 8, 8])
run(atoms, 'manykpts')
@pytest.mark.skip('expensive')
@calc('abinit')
def test_manyatoms(factory):
atoms = bulk('Ne', cubic=True) * (4, 2, 2)
atoms.rattle(stdev=0.01)
atoms.calc = factory.calc(nbands=len(atoms) * 5)
run(atoms, 'manyatoms')
|
