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
|
def test_surface_stack():
from ase.build.surface import _all_surface_functions
from ase.build import stack
from ase.calculators.calculator import compare_atoms
# The purpose of this test is to test the stack() function and verify
# that the various surface builder functions produce configurations
# consistent with stacking.
d = _all_surface_functions()
exclude = {'mx2', 'graphene'} # mx2 and graphene are not like the others
for name in sorted(d):
if name in exclude:
continue
func = d[name]
def has(var):
c = func.__code__
return var in c.co_varnames[:c.co_argcount]
for nlayers in range(1, 7):
atoms = func('Au', size=(2, 2, nlayers), periodic=True, a=4.0)
big_atoms = func('Au', size=(2, 2, 2 * nlayers), periodic=True, a=4.0)
stacked_atoms = stack(atoms, atoms)
changes = compare_atoms(stacked_atoms, big_atoms, tol=1e-11)
if not changes:
print('OK', name, nlayers)
break
else:
assert 0, 'Unstackable surface {}'.format(name)
|
