def test2(): from Atom import Atom from Molecule import Molecule from TullySurfaceHopping import TullySurfaceHopping from Constants import fs2tau, tau2fs from IO_MOLPRO import molpro_input_parser from VelocityVerlet import set_random_momentum mol, inp = molpro_input_parser("mlpinp1.com") set_random_momentum(mol,0.1) pot = Potential_TSH(mol,inp,now_state=0,nrange=3) pot.override_output("mlpinp0.out_01") read_velocity(mol,"velocity5_1") read_coord(mol,"coord5_1") tsh = TullySurfaceHopping(mol,pot,dt=0.5*fs2tau,nstep=100,\ tsh_times=5) # These are special fort TSH tsh.run()