def test_hybridcloud_pkl_interface(): hc = test_hybridcloud_load() fname = f'{test_dir}/test_hc.pkl' try: hc.save2pkl(fname) hc2 = HybridCloud() hc2.load_from_pkl(fname) assert hc == hc2 finally: if os.path.isfile(fname): os.remove(fname)
def load_obj( data_type: str, file: str ) -> Union[HybridMesh, HybridCloud, PointCloud, CloudEnsemble]: if data_type == 'obj': return basics.load_pkl(file) if data_type == 'ce': return ensembles.ensemble_from_pkl(file) if data_type == 'hc': hc = HybridCloud() return hc.load_from_pkl(file) if data_type == 'hm': hm = HybridMesh() return hm.load_from_pkl(file) else: pc = PointCloud() return pc.load_from_pkl(file)