def test_pickle_manipulation(self, testdir, n2_emt):
atoms = n2_emt
vib = Vibrations(atoms, name='interrupt')
vib.run()
disp_file = 'interrupt/cache.1x-.json'
comb_file = 'interrupt/combined.json'
assert os.path.isfile(disp_file)
assert not os.path.isfile(comb_file)
#with pytest.raises(RuntimeError):
# vib.split()
# Build a combined file
assert vib.combine() == 13
# Individual displacements should be gone, combination should exist
assert not os.path.isfile(disp_file)
assert os.path.isfile(comb_file)
# Not allowed to run after data has been combined
with pytest.raises(RuntimeError):
vib.run()
# But reading is allowed
vib.read()
# Splitting should fail if any split file already exists
with open(disp_file, 'w') as fd:
fd.write("hello")
#with pytest.raises(RuntimeError):
# vib.split()
os.remove(disp_file)
# Now split() for real: replace .all.json file with displacements
vib.split()
assert os.path.isfile(disp_file)
assert not os.path.isfile(comb_file)
# Not allowed to clobber existing combined file
with open(comb_file, 'w') as fd:
fd.write("Hello")
#with pytest.raises(RuntimeError):
# vib.combine()
os.remove(comb_file)
# Combining data also fails if some data is missing
os.remove('interrupt/cache.1x-.json')
def test_vib():
import os
from ase import Atoms
from ase.calculators.emt import EMT
from ase.optimize import QuasiNewton
from ase.vibrations import Vibrations
from ase.thermochemistry import IdealGasThermo
n2 = Atoms('N2', positions=[(0, 0, 0), (0, 0, 1.1)], calculator=EMT())
QuasiNewton(n2).run(fmax=0.01)
vib = Vibrations(n2)
vib.run()
freqs = vib.get_frequencies()
print(freqs)
vib.summary()
print(vib.get_mode(-1))
vib.write_mode(n=None, nimages=20)
vib_energies = vib.get_energies()
for image in vib.iterimages():
assert len(image) == 2
thermo = IdealGasThermo(vib_energies=vib_energies,
geometry='linear',
atoms=n2,
symmetrynumber=2,
spin=0)
thermo.get_gibbs_energy(temperature=298.15, pressure=2 * 101325.)
assert vib.clean(empty_files=True) == 0
assert vib.clean() == 13
assert len(list(vib.iterimages())) == 13
d = dict(vib.iterdisplace(inplace=False))
for name, atoms in vib.iterdisplace(inplace=True):
assert d[name] == atoms
vib = Vibrations(n2)
vib.run()
assert vib.combine() == 13
assert (freqs == vib.get_frequencies()).all()
vib = Vibrations(n2)
assert vib.split() == 1
assert (freqs == vib.get_frequencies()).all()
assert vib.combine() == 13
# Read the data from other working directory
dirname = os.path.basename(os.getcwd())
os.chdir('..') # Change working directory
vib = Vibrations(n2, name=os.path.join(dirname, 'vib'))
assert (freqs == vib.get_frequencies()).all()
assert vib.clean() == 1
def test_json_manipulation(self, testdir, random_dimer):
vib = Vibrations(random_dimer, name='interrupt')
vib.run()
disp_file = Path('interrupt/cache.1x-.json')
comb_file = Path('interrupt/combined.json')
assert disp_file.is_file()
assert not comb_file.is_file()
# Should do nothing harmful as files are already split
# (It used to raise an error but this is no longer implemented.)
vib.split()
# Build a combined file
assert vib.combine() == 13
# Individual displacements should be gone, combination should exist
assert not disp_file.is_file()
assert comb_file.is_file()
# Not allowed to run after data has been combined
with pytest.raises(RuntimeError):
vib.run()
# But reading is allowed
vib.read()
# Splitting should fail if any split file already exists
with open(disp_file, 'w') as fd:
fd.write("hello")
with pytest.raises(AssertionError):
vib.split()
os.remove(disp_file)
# Now split() for real: replace .all.json file with displacements
vib.split()
assert disp_file.is_file()
assert not comb_file.is_file()
for image in vib.iterimages():
assert len(image) == 2
thermo = IdealGasThermo(vib_energies=vib_energies,
geometry='linear',
atoms=n2,
symmetrynumber=2,
spin=0)
thermo.get_gibbs_energy(temperature=298.15, pressure=2 * 101325.)
assert vib.clean(empty_files=True) == 0
assert vib.clean() == 13
assert len(list(vib.iterimages())) == 13
d = dict(vib.iterdisplace(inplace=False))
for name, atoms in vib.iterdisplace(inplace=True):
assert d[name] == atoms
vib = Vibrations(n2)
vib.run()
assert vib.combine() == 13
assert (freqs == vib.get_frequencies()).all()
vib = Vibrations(n2)
assert vib.split() == 1
assert (freqs == vib.get_frequencies()).all()
assert vib.combine() == 13
assert vib.clean() == 1
