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)