def test_ori_from_particles(self): """Orientation preparation in ori from particles.""" # Note: this result set isn't great for testing shaking - it is only # used for testing the prepare_eval part! shutil.copytree("db_targ/", "scene83_event1/") shutil.copytree("shaking_res/", "res/") shutil.copyfile("parameters/sequence.par", "parameters/sequence_scene.par") shutil.copyfile("parameters/sequence_db.par", "parameters/sequence.par") if os.path.exists("cal_bk/"): shutil.rmtree("cal_bk/") shutil.copytree("cal/", "cal_bk/") py_init_proc_c() py_start_proc_c() py_calibration(10) pix, crd = get_pix_crd(4) #np.savetxt('shaking_res/pix_regress.dat', pix.reshape(-1, 2)) #np.savetxt('shaking_res/crd_regress.dat', crd.reshape(-1, 3)) np.testing.assert_array_almost_equal( pix.reshape(-1, 2), np.loadtxt('shaking_res/pix_regress.dat')) np.testing.assert_array_almost_equal( crd.reshape(-1, 3), np.loadtxt('shaking_res/crd_regress.dat')) shutil.rmtree("cal/") shutil.copytree("cal_bk/", "cal/") shutil.rmtree("cal_bk/") for file in glob.glob("safety_*"): os.remove(file)
def test_ori_from_particles(self): """Orientation preparation in ori from particles.""" # Note: this result set isn't great for testing shaking - it is only # used for testing the prepare_eval part! shutil.copytree("db_targ/", "scene83_event1/") shutil.copytree("shaking_res/", "res/") shutil.copyfile("parameters/sequence.par", "parameters/sequence_scene.par") shutil.copyfile("parameters/sequence_db.par", "parameters/sequence.par") if os.path.exists("cal_bk/"): shutil.rmtree("cal_bk/") shutil.copytree("cal/", "cal_bk/") py_init_proc_c() py_start_proc_c() py_calibration(10) pix, crd = get_pix_crd(4) #np.savetxt('shaking_res/pix_regress.dat', pix.reshape(-1, 2)) #np.savetxt('shaking_res/crd_regress.dat', crd.reshape(-1, 3)) np.testing.assert_array_almost_equal(pix.reshape(-1, 2), np.loadtxt('shaking_res/pix_regress.dat')) np.testing.assert_array_almost_equal(crd.reshape(-1, 3), np.loadtxt('shaking_res/crd_regress.dat')) shutil.rmtree("cal/") shutil.copytree("cal_bk/", "cal/") shutil.rmtree("cal_bk/") for file in glob.glob("safety_*"): os.remove(file)
def test_raw_orient(self): """Check that raw_orient doesn't ruin the results for just_plot().""" shutil.copytree("db_targ/", "scene83_event1/") py_init_proc_c() py_start_proc_c() py_calibration(9) calib = get_xy_calib(4) #np.savetxt('calib.dat', calib.reshape(-1, 2)) np.testing.assert_array_almost_equal(calib.reshape(-1, 2), np.loadtxt('calib.dat')) for file in glob.glob("safety_*"): os.remove(file)