def main():
"""Simply merge all trajectories in the working directory"""
folder = os.getcwd()
print('Merging all files')
merge_all_in_folder(folder,
delete_other_files=True, # We will only keep one trajectory
dynamic_imports=FunctionParameter,
backup=False)
print('Done')
def main():
"""Simply merge all trajectories in the working directory"""
folder = os.getcwd()
print('Merging all files')
merge_all_in_folder(
folder,
delete_other_files=True, # We will only keep one trajectory
dynamic_imports=FunctionParameter,
backup=False)
print('Done')
def test_merge_all_in_folder(self):
self.filename = make_temp_dir(os.path.join('experiments','tests','HDF5', 'subfolder',
'test.hdf5'))
path, _ = os.path.split(self.filename)
ntrajs = 4
total_len = 0
for irun in range(ntrajs):
new_filename = os.path.join(path, 'test%d.hdf5' % irun)
self.envs.append(self._make_env(irun, filename=new_filename))
self.trajs.append(self.envs[-1].v_traj)
self.trajs[-1].f_add_parameter('x',0)
self.trajs[-1].f_add_parameter('y',0)
self.explore(self.trajs[-1])
total_len += len(self.trajs[-1])
for irun in range(ntrajs):
self.envs[irun].f_run(multiply)
merge_traj = merge_all_in_folder(path, delete_other_files=True)
merge_traj.f_load(load_data=2)
self.assertEqual(len(merge_traj), total_len)
self.check_if_z_is_correct(merge_traj)
