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
|
import pytest
import numpy as np
from ase.outputs import Properties, all_outputs
@pytest.fixture
def rng():
return np.random.RandomState(17)
def test_properties_big(rng):
nspins, nkpts, nbands = 2, 3, 5
natoms = 4
results = dict(
natoms=natoms,
energy=rng.rand(),
free_energy=rng.rand(),
energies=rng.rand(natoms),
forces=rng.rand(natoms, 3),
stress=rng.rand(6),
stresses=rng.rand(natoms, 6),
nspins=nspins,
nkpts=nkpts,
nbands=nbands,
eigenvalues=rng.rand(nspins, nkpts, nbands),
occupations=rng.rand(nspins, nkpts, nbands),
fermi_level=rng.rand(),
ibz_kpoints=rng.rand(nkpts, 3),
kpoint_weights=rng.rand(nkpts),
)
props = Properties(results)
#assert set(out) == all_outputs, all_outputs ^ set(out)
for name in all_outputs:
assert name in props, name
obj = props[name]
#obj = getattr(out, name)
print(name, obj)
|
