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
|
import numpy as np
from ase.build import bulk
from ase.calculators.singlepoint import (
SinglePointDFTCalculator,
arrays_to_kpoints,
)
def test_singlepoint_dft_calc():
rng = np.random.RandomState(17)
nspins, nkpts, _nbands = shape = 2, 4, 5
eps = 2 * rng.random(shape)
occ = rng.random(shape)
weights = rng.random(nkpts)
kpts = arrays_to_kpoints(eps, occ, weights)
atoms = bulk('Au')
calc = SinglePointDFTCalculator(atoms)
calc.kpts = kpts
assert calc.get_number_of_spins() == nspins
assert calc.get_spin_polarized()
assert np.allclose(calc.get_k_point_weights(), weights)
for s in range(nspins):
for k in range(nkpts):
eps1 = calc.get_eigenvalues(kpt=k, spin=s)
occ1 = calc.get_occupation_numbers(kpt=k, spin=s)
assert np.allclose(eps1, eps[s, k])
assert np.allclose(occ1, occ[s, k])
# XXX Should check more stuff.
|
