def test_filesetup():
""" Tests file setup. """
from tempfile import mkdtemp
from os.path import exists, join, samefile, lexists
from shutil import rmtree
from pylada.gulp import Functional
from pylada.crystal import Structure
from pylada.tools import create_directory, get_section_from_file
functional = Functional()
functional.morse.enabled = True
functional.morse['Ti', 'O'] = [1.0279493, 3.640737, 1.88265, 0, 25]
functional.morse['Ti', 'Ti'] = [0.00567139, 1.5543, 4.18784, 0, 25]
functional.morse['O', 'O'] = [0.00567139, 1.5543, 4.18784, 0, 25]
structure = Structure( 4.6391, 0, 0,
0, 4.6391, 0,
0, 0, 2.97938,
scale=1, symmgroup=136 ) \
.add_atom(0, 0, 0, 'Ti', asymmetric=True) \
.add_atom(2.31955, 2.31955, 1.48969, 'Ti', asymmetric=False) \
.add_atom(1.42027, 1.42027, 0, 'O', asymmetric=True) \
.add_atom(-1.42027, -1.42027, 0, 'O', asymmetric=False) \
.add_atom(-0.899275, 0.899275, 1.48969, 'O', asymmetric=False) \
.add_atom(0.899275, -0.899275, 1.48969, 'O', asymmetric=False)
directory = mkdtemp()
if directory == '/tmp/test/':
if exists(directory): rmtree(directory)
create_directory(directory)
try:
iterator = functional.iter(structure, outdir=directory)
assert exists(directory)
assert not exists(join(directory, 'gulp.out'))
assert not exists(join(directory, 'gulp.in'))
assert not exists(join(directory, 'gulp.err'))
program = iterator.next()
assert exists(join(directory, 'gulp.out'))
assert exists(join(directory, 'gulp.in'))
assert not exists(join(directory, 'gulp.err'))
with open(join(directory, 'gulp.out'), 'r') as file:
get_section_from_file(file, 'input file')
get_section_from_file(file, 'functional')
rmtree(directory)
create_directory(directory)
workdir = join(directory, 'wtf')
iterator = functional.iter( structure, outdir=directory,
workdir=workdir )
assert exists(directory)
assert not exists(join(directory, 'gulp.out'))
assert not exists(join(directory, 'gulp.in'))
assert not exists(join(directory, 'gulp.err'))
program = iterator.next()
assert exists(join(directory, 'gulp.out'))
assert exists(join(directory, 'gulp.in'))
assert not exists(join(directory, 'gulp.err'))
with open(join(directory, 'gulp.out'), 'r') as file:
get_section_from_file(file, 'input file')
get_section_from_file(file, 'functional')
assert samefile(workdir, join(directory, 'workdir'))
assert exists(join(workdir, 'gulp.out'))
assert samefile(join(workdir, 'gulp.out'), join(directory, 'gulp.out'))
assert exists(join(workdir, 'gulp.in'))
assert not samefile(join(workdir, 'gulp.in'), join(directory, 'gulp.in'))
assert lexists(join(workdir, 'gulp.err'))
assert exists(join(directory, '.pylada_is_running'))
program.onfinish()
assert not exists(join(directory, '.pylada_is_running'))
assert exists(workdir)
assert exists(join(directory, 'workdir'))
assert samefile(join(workdir, 'gulp.out'), join(directory, 'gulp.out'))
finally:
if directory != '/tmp/test/': rmtree(directory)
