def test_intermittent(self): hbond = HBAC(self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='intermittent', sample_time=0.06, ) hbond.run() assert_array_almost_equal( hbond.solution['results'], np.array([ 1. , 0.92668623, 0.84310848, 0.79325515, 0.76392961, 0.72287393], dtype=np.float32) )
def test_save(self): hbond = HBAC(self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='continuous', sample_time=0.06, ) hbond.run() with tempdir.in_tempdir(): hbond.save_results('hbondout.npz') loaded = np.load('hbondout.npz') assert_('time' in loaded) assert_('results' in loaded)
def test_continuous(self): hbond = HBAC(self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='continuous', sample_time=0.06, ) hbond.run() assert_array_almost_equal( hbond.solution['results'], np.array([ 1. , 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932], dtype=np.float32) )
def test_intermittent(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC( u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='intermittent', sample_time=0.06, ) hbond.run() assert_almost_equal( hbond.solution['results'], np.array([ 1., 0.92668623, 0.84310848, 0.79325515, 0.76392961, 0.72287393 ], dtype=np.float32))
def test_continuous(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC( u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='continuous', sample_time=0.06, ) hbond.run() assert_almost_equal( hbond.solution['results'], np.array([ 1., 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932 ], dtype=np.float32))
def test_save(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC( u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='continuous', sample_time=0.06, ) hbond.run() with tempdir.in_tempdir(): hbond.save_results('hbondout.npz') loaded = np.load('hbondout.npz') assert 'time' in loaded assert 'results' in loaded
def test_save(self): hbond = HBAC( self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='continuous', sample_time=0.06, ) hbond.run() with tempdir.in_tempdir(): hbond.save_results('hbondout.npz') loaded = np.load('hbondout.npz') assert_('time' in loaded) assert_('results' in loaded)
def test_intermittent_timecut(self): hbond = HBAC(self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='intermittent', time_cut=0.01, # time cut at traj.dt == continuous sample_time=0.06, ) hbond.run() assert_array_almost_equal( hbond.solution['results'], np.array([ 1. , 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932], dtype=np.float32) )
def test_save(self, u, hydrogens, oxygens, nitrogens, tmpdir): hbond = HBAC(u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='continuous', sample_time=0.06, ) hbond.run() tmpfile = os.path.join(str(tmpdir), 'hbondout.npz') hbond.save_results(tmpfile) loaded = np.load(tmpfile) assert 'time' in loaded assert 'results' in loaded
def test_intermittent_timecut(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC( u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='intermittent', time_cut=0.01, # time cut at traj.dt == continuous sample_time=0.06, ) hbond.run() assert_almost_equal( hbond.solution['results'], np.array([ 1., 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932 ], dtype=np.float32))
def test_continuous_excl(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC(u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='continuous', exclusions=(np.arange(len(hydrogens)), np.array( range(len(oxygens)))), sample_time=0.06, ) hbond.run() assert_almost_equal( hbond.solution['results'], np.array([ 1. , 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932], dtype=np.float32) )
def test_intermittent_excl(self, u, hydrogens, oxygens, nitrogens): hbond = HBAC(u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='intermittent', exclusions=(np.arange(len(hydrogens)), np.array( range(len(oxygens)))), sample_time=0.06, ) hbond.run() assert_almost_equal( hbond.solution['results'], np.array([ 1. , 0.92668623, 0.84310848, 0.79325515, 0.76392961, 0.72287393], dtype=np.float32) )
def test_save(self, u, hydrogens, oxygens, nitrogens, tmpdir): hbond = HBAC( u, hydrogens=hydrogens, acceptors=oxygens, donors=nitrogens, bond_type='continuous', sample_time=0.06, ) hbond.run() tmpfile = os.path.join(str(tmpdir), 'hbondout.npz') hbond.save_results(tmpfile) loaded = np.load(tmpfile) assert 'time' in loaded assert 'results' in loaded
def test_continuous_excl(self): hbond = HBAC( self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='continuous', exclusions=self.excl_list, sample_time=0.06, ) hbond.run() assert_array_almost_equal( hbond.solution['results'], np.array([ 1., 0.92668623, 0.83137828, 0.74486804, 0.67741936, 0.60263932 ], dtype=np.float32))
def test_intermittent_excl(self): hbond = HBAC( self.u, hydrogens=self.H, acceptors=self.O, donors=self.N, bond_type='intermittent', exclusions=self.excl_list, sample_time=0.06, ) hbond.run() assert_array_almost_equal( hbond.solution['results'], np.array([ 1., 0.92668623, 0.84310848, 0.79325515, 0.76392961, 0.72287393 ], dtype=np.float32))
'O9' : u.select_atoms('bynum 64'), 'O10' : u.select_atoms('bynum 18') } oxygens = sum(list(oxygen_dict.values())) hydrogens = sum(list(hydrogen_dict.values())) nitrogens = sum(list(nitrogen_dict.values())) "" "Run HBond Autocorrelation" hb_ac = HydrogenBondAutoCorrel(u, acceptors = oxygens, hydrogens = hydrogens, donors = nitrogens, bond_type='continuous') hb_ac.run() hb_ac.solve() tau = hb_ac.solution['tau'] time = hb_ac.solution['time'] results = hb_ac.solution['results'] estimate = hb_ac.solution['estimate'] "Max Range" factor = 0.5 max_d = int(data[:,0].size * factor) max_h = int(time.size * factor) max_r = int(len(u.trajectory) * factor) "Plot Results and Find Correlation" fig, ax1 = plt.subplots() color = 'red'