"""
Unit and regression test for utilities.py.
"""
# Import package, test suite, and other packages as needed
import atesa
import sys
import os
import shutil
import pytest
import pickle
import subprocess
import glob
from atesa import rc_eval
from atesa.configure import configure
from atesa import main
[docs]class Tests(object):
[docs] def setup_method(self, test_method):
try:
if not os.path.exists('atesa/tests/test_temp'):
os.mkdir('atesa/tests/test_temp')
os.chdir('atesa/tests/test_temp')
except FileNotFoundError:
pass
[docs] def test_main_pytraj(self):
"""Tests rc_eval.main using sham shooting points in test_data and pytraj cvs"""
settings = configure('../../data/atesa.config')
settings.topology = '../test_data/test.prmtop'
settings.cvs = ['pytraj.distance(traj, \'@1 @2\')[0]', 'pytraj.angle(traj, \'@2 @3 @4\')[0]']
settings.include_qdot = False
shutil.copy('../test_data/test_velocities_init.rst7', '../test_temp')
shutil.copy('../test_data/test_two_init.rst7', '../test_temp')
shutil.copy('../test_data/as.out', '../test_temp')
# First test without extant settings.pkl file
with pytest.raises(FileNotFoundError):
rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out')
process = subprocess.Popen('rc_eval.py ../test_temp/ 3*CV0-0.3*CV1 ../test_data/as.out', stdin=subprocess.PIPE,
stdout=subprocess.PIPE, stderr=subprocess.STDOUT, close_fds=True, shell=True)
output = process.stdout.read().decode()
assert 'FileNotFoundError' in output # this section just to ensure it works when called both ways
settings.__dict__.pop('env') # env attribute is not picklable
pickle.dump(settings, open('settings.pkl', 'wb'), protocol=2) # main will look for this file to load in settings
# shutil.move('settings.pkl', '../test_data/settings.pkl')
rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out')
assert os.path.exists('rc.out')
lines = open('rc.out', 'r').readlines()
for i in range(len(lines)): # assert that the rc.out file is properly sorted
try:
assert abs(float(lines[i].split(': ')[1])) < abs(float(lines[i+1].split(': ')[1]))
except IndexError:
pass
[docs] def test_main_mdtraj(self):
"""Tests rc_eval.main using sham shooting points in test_data and mdtraj cvs"""
settings = configure('../../data/atesa.config')
settings.topology = '../test_data/test.prmtop'
settings.cvs = ['mdtraj.compute_distances(mtraj, numpy.array([[0, 1]]))[0][0] * 10', 'mdtraj.compute_angles(mtraj, numpy.array([[1, 2, 3]]))[0][0] * 180 / numpy.pi']
settings.include_qdot = False
shutil.copy('../test_data/test_velocities_init.rst7', '../test_temp')
shutil.copy('../test_data/test_two_init.rst7', '../test_temp')
shutil.copy('../test_data/as.out', '../test_temp')
# First test without extant settings.pkl file
with pytest.raises(FileNotFoundError):
rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out')
process = subprocess.Popen('rc_eval.py ../test_temp/ 3*CV0-0.3*CV1 ../test_data/as.out', stdin=subprocess.PIPE,
stdout=subprocess.PIPE, stderr=subprocess.STDOUT, close_fds=True, shell=True)
output = process.stdout.read().decode()
assert 'FileNotFoundError' in output # this section just to ensure it works when called both ways
settings.__dict__.pop('env') # env attribute is not picklable
pickle.dump(settings, open('settings.pkl', 'wb'), protocol=2) # main will look for this file to load in settings
# shutil.move('settings.pkl', '../test_data/settings.pkl')
rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out')
assert os.path.exists('rc.out')
lines = open('rc.out', 'r').readlines()
for i in range(len(lines)): # assert that the rc.out file is properly sorted
try:
assert abs(float(lines[i].split(': ')[1])) < abs(float(lines[i+1].split(': ')[1]))
except IndexError:
pass
[docs] def test_extrema_pytraj(self):
"""Tests rc_eval.main with extrema=True using sham shooting points in test_data and pytraj cvs"""
settings = configure('../../data/atesa.config')
settings.topology = '../test_data/ethyl_chlorosulfite.prmtop'
settings.cvs = ['pytraj.distance(traj, \'@1 @7\')[0]', 'pytraj.angle(traj, \'@2 @6 @9\')[0]']
settings.include_qdot = False
shutil.copy('../test_data/test_velocities_init.rst7', '../test_temp')
shutil.copy('../test_data/test_two_init.rst7', '../test_temp')
shutil.copy('../test_data/as.out', '../test_temp')
shutil.copy('../test_data/init_-0.20000000000000018_0.rst7_0_us.nc', '../test_temp')
allthreads = [main.init_threads(settings)[0]]
allthreads[0].history.prod_trajs = [['init_-0.20000000000000018_0.rst7_0_us.nc', 'init_-0.20000000000000018_0.rst7_0_us.nc']]
allthreads[0].history.init_coords = ['test_velocities_init.rst7']
allthreads[0].history.last_accepted = 0
pickle.dump(allthreads, open('restart.pkl', 'wb'))
settings.__dict__.pop('env') # env attribute is not picklable
pickle.dump(settings, open('settings.pkl', 'wb'),
protocol=2) # main will look for this file to load in settings
# shutil.move('settings.pkl', '../test_data/settings.pkl')
extrema = rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out', extrema=True)
assert extrema == [71.4222, 71.4222]
[docs] def test_extrema_mdtraj(self):
"""Tests rc_eval.main with extrema=True using sham shooting points in test_data and pytraj cvs"""
settings = configure('../../data/atesa.config')
settings.topology = '../test_data/ethyl_chlorosulfite.prmtop'
settings.cvs = ['mdtraj.compute_distances(mtraj, numpy.array([[0, 6]]))[0][0] * 10', 'mdtraj.compute_angles(mtraj, numpy.array([[1, 5, 8]]))[0][0] * 180 / numpy.pi']
settings.include_qdot = False
shutil.copy('../test_data/test_velocities_init.rst7', '../test_temp')
shutil.copy('../test_data/test_two_init.rst7', '../test_temp')
shutil.copy('../test_data/as.out', '../test_temp')
shutil.copy('../test_data/init_-0.20000000000000018_0.rst7_0_us.nc', '../test_temp')
allthreads = [main.init_threads(settings)[0]]
allthreads[0].history.prod_trajs = [['init_-0.20000000000000018_0.rst7_0_us.nc', 'init_-0.20000000000000018_0.rst7_0_us.nc']]
allthreads[0].history.init_coords = ['test_velocities_init.rst7']
allthreads[0].history.last_accepted = 0
pickle.dump(allthreads, open('restart.pkl', 'wb'))
settings.__dict__.pop('env') # env attribute is not picklable
pickle.dump(settings, open('settings.pkl', 'wb'),
protocol=2) # main will look for this file to load in settings
# shutil.move('settings.pkl', '../test_data/settings.pkl')
extrema = rc_eval.main('../test_temp/', '3*CV1 - 0.3*CV2', '../test_data/as.out', extrema=True)
assert extrema == [71.4222, 71.4222]
[docs] @classmethod
def teardown_method(self, method):
"Runs at end of class"
for filename in glob.glob(sys.path[0] + '/atesa/tests/test_temp/*'):
os.remove(filename)