def test_lfp(self): # set config for unit test purposes cfg = PlenopticamConfig() cfg.params[cfg.opt_dbug] = True for fn_lfp, fn_wht in zip(self.fnames_lfp, self.fnames_wht): # update file paths and calibration data in config cfg.params[cfg.lfp_path] = os.path.join(self.fp, fn_lfp) cfg.params[cfg.cal_path] = os.path.join(self.fp, fn_wht) cfg.params[cfg.cal_meta] = os.path.splitext( cfg.params[cfg.cal_path])[0] + '.json' cfg.load_cal_data() # create folder (if it doesn't already exist) misc.mkdir_p(os.path.splitext(cfg.params[cfg.lfp_path])[0]) # test light field alignment lfp_img = misc.load_img_file(cfg.params[cfg.lfp_path]) lfp_obj = LfpAligner(lfp_img=lfp_img, cfg=cfg, sta=None) ret_val = lfp_obj.main() lfp_img = lfp_obj.lfp_img del lfp_obj # assertion self.assertEqual(True, ret_val) # test light field extraction lfp_obj = LfpExtractor(lfp_img_align=lfp_img, cfg=cfg, sta=None) ret_val = lfp_obj.main() del lfp_obj # assertion self.assertEqual(True, ret_val)
def test_lfp(self): # set config for unit test purposes sta = PlenopticamStatus() cfg = PlenopticamConfig() cfg.reset_values() cfg.params[cfg.opt_dbug] = False cfg.params[ cfg. opt_prnt] = False # prevent Travis CI to terminate after reaching 4MB logfile size for fn_lfp, fn_wht in zip(self.fnames_lfp, self.fnames_wht): # generate console output to prevent abort in Travis CI print(fn_lfp) # update file paths and calibration data in config cfg.params[cfg.lfp_path] = os.path.join(self.fp, fn_lfp) cfg.params[cfg.cal_path] = os.path.join(self.fp, fn_wht) cfg.params[cfg.cal_meta] = os.path.splitext( cfg.params[cfg.cal_path])[0] + '.json' cfg.load_cal_data() # create folder (if it doesn't already exist) mkdir_p(os.path.splitext(cfg.params[cfg.lfp_path])[0]) # test light field alignment lfp_img = load_img_file(cfg.params[cfg.lfp_path]) lfp_obj = LfpAligner(lfp_img=lfp_img, cfg=cfg, sta=sta) ret_val = lfp_obj.main() lfp_img = lfp_obj.lfp_img del lfp_obj # assertion self.assertEqual(True, ret_val) # test light field extraction lfp_obj = LfpExtractor(lfp_img_align=lfp_img, cfg=cfg, sta=sta) lfp_obj.main() vp_img_arr = lfp_obj.vp_img_arr del lfp_obj lfp_obj = LfpRefocuser(vp_img_arr=vp_img_arr, cfg=cfg, sta=sta) lfp_obj.main() del lfp_obj # assertion self.assertEqual(True, ret_val)
def test_custom_lfp(self): for fn_lfp, fn_wht in zip(self.loader.opex_fnames_lfp, self.loader.opex_fnames_wht): # generate console output to prevent abort in Travis CI print(fn_lfp) # update file paths and calibration data in config self.cfg.params[self.cfg.lfp_path] = join(self.fp, fn_lfp) self.cfg.params[self.cfg.cal_path] = join(self.fp, fn_wht) self.cfg.params[self.cfg.cal_meta] = splitext( self.cfg.params[self.cfg.cal_path])[0] + '.json' self.cfg.load_cal_data() # create folder (if it doesn't already exist) mkdir_p(splitext(self.cfg.params[self.cfg.lfp_path])[0]) # test light field alignment lfp_img = load_img_file(self.cfg.params[self.cfg.lfp_path]) lfp_obj = LfpAligner(lfp_img=lfp_img, cfg=self.cfg, sta=self.sta) ret_val = lfp_obj.main() lfp_img = lfp_obj.lfp_img del lfp_obj # assertion self.assertEqual(True, ret_val) # test light field extraction lfp_obj = LfpExtractor(lfp_img_align=lfp_img, cfg=self.cfg, sta=self.sta) ret_val = lfp_obj.main() vp_img_arr = lfp_obj.vp_img_arr del lfp_obj # assertion self.assertEqual(True, ret_val) lfp_obj = LfpRefocuser(vp_img_arr=vp_img_arr, cfg=self.cfg, sta=self.sta) ret_val = lfp_obj.main() del lfp_obj # assertion self.assertEqual(True, ret_val)