def test_get_dec_and_inc(self): # testing full thing with real data
dec, inc, intenstiy, tau, V, means, PCA_sigma = lib_direct.get_dec_and_inc(
PintPars_example.zdata,
PintPars_example.t_Arai,
PintPars_example.tmin,
PintPars_example.tmax,
anchored=False)
self.assertAlmostEqual(dec, 267.4620127216387)
self.assertAlmostEqual(inc, 86.349431762792364)
self.assertGreaterEqual(tau[0], tau[1])
self.assertGreaterEqual(tau[1], tau[2])
def get_dec_and_inc(self):
Dec_Anc, Inc_Anc, best_fit_Anc, tau_Anc, V_Anc, mass_center, PCA_sigma_Anc = lib_direct.get_dec_and_inc(self.zdata,
self.t_Arai, self.tmin, self.tmax, anchored=True)
Dec_Free, Inc_Free, best_fit_Free, tau_Free, V_Free, mass_center, PCA_sigma_Free = lib_direct.get_dec_and_inc(self.zdata,
self.t_Arai, self.tmin, self.tmax, anchored=False)
self.pars['Dec_Anc'], self.pars['Dec_Free'] = Dec_Anc, Dec_Free
self.pars['Inc_Anc'], self.pars['Inc_Free'] = Inc_Anc, Inc_Free
self.pars['best_fit_vector_Anc'], self.pars['best_fit_vector_Free'] = best_fit_Anc, best_fit_Free
self.pars['tau_Anc'], self.pars['tau_Free'] = tau_Anc, tau_Free
self.pars['V_Anc'], self.pars['V_Free'] = V_Anc, V_Free
self.pars['zdata_mass_center'] = mass_center
self.pars['PCA_sigma_max_Free'] = PCA_sigma_Free[0]
self.pars['PCA_sigma_int_Free'] = PCA_sigma_Free[1]
self.pars['PCA_sigma_min_Free'] = PCA_sigma_Free[2]
def get_dec_and_inc(self):
Dec_Anc, Inc_Anc, best_fit_Anc, tau_Anc, V_Anc, mass_center, PCA_sigma_Anc = lib_direct.get_dec_and_inc(self.zdata,
self.t_Arai, self.tmin, self.tmax, anchored=True)
Dec_Free, Inc_Free, best_fit_Free, tau_Free, V_Free, mass_center, PCA_sigma_Free = lib_direct.get_dec_and_inc(self.zdata,
self.t_Arai, self.tmin, self.tmax, anchored=False)
self.pars['Dec_Anc'], self.pars['Dec_Free'] = Dec_Anc, Dec_Free
self.pars['Inc_Anc'], self.pars['Inc_Free'] = Inc_Anc, Inc_Free
self.pars['best_fit_vector_Anc'] = best_fit_Anc
self.pars['best_fit_vector_Free'] = best_fit_Free
self.pars['tau_Anc'], self.pars['tau_Free'] = tau_Anc, tau_Free
self.pars['V_Anc'], self.pars['V_Free'] = V_Anc, V_Free
self.pars['zdata_mass_center'] = mass_center
self.pars['PCA_sigma_max_Free'] = PCA_sigma_Free[0]
self.pars['PCA_sigma_int_Free'] = PCA_sigma_Free[1]
self.pars['PCA_sigma_min_Free'] = PCA_sigma_Free[2]
def get_ptrm_dec_and_inc(self):
"""not included in spd."""
PTRMS = self.PTRMS[1:]
CART_pTRMS_orig = numpy.array([lib_direct.dir2cart(row[1:4]) for row in PTRMS])
#B_lab_dir = [self.B_lab_dir[0], self.B_lab_dir[1], 1.] # dir
tmin, tmax = self.t_Arai[0], self.t_Arai[-1]
ptrms_dec_Free, ptrms_inc_Free, ptrm_best_fit_vector_Free, ptrm_tau_Free, ptrm_v_Free, ptrm_mass_center_Free, ptrm_PCA_sigma_Free = lib_direct.get_dec_and_inc(CART_pTRMS_orig, self.t_Arai, tmin, tmax, anchored=False)
ptrms_angle = lib_direct.get_ptrms_angle(ptrm_best_fit_vector_Free, self.B_lab_cart)
self.pars['ptrms_dec_Free'], self.pars['ptrms_inc_Free'] = ptrms_dec_Free, ptrms_inc_Free
self.pars['ptrms_tau_Free'] = ptrm_tau_Free
self.pars['ptrms_angle_Free'] = ptrms_angle
def get_ptrm_dec_and_inc(self):
"""not included in spd."""
PTRMS = self.PTRMS[1:]
CART_pTRMS_orig = numpy.array([lib_direct.dir2cart(row[1:4]) for row in PTRMS])
#B_lab_dir = [self.B_lab_dir[0], self.B_lab_dir[1], 1.] # dir
tmin, tmax = self.t_Arai[0], self.t_Arai[-1]
ptrms_dec_Free, ptrms_inc_Free, ptrm_best_fit_vector_Free, ptrm_tau_Free, ptrm_v_Free, ptrm_mass_center_Free, ptrm_PCA_sigma_Free = lib_direct.get_dec_and_inc(CART_pTRMS_orig, self.t_Arai, tmin, tmax, anchored=False)
ptrms_angle = lib_direct.get_ptrms_angle(ptrm_best_fit_vector_Free, self.B_lab_cart)
self.pars['ptrms_dec_Free'], self.pars['ptrms_inc_Free'] = ptrms_dec_Free, ptrms_inc_Free
self.pars['ptrms_tau_Free'] = ptrm_tau_Free
self.pars['ptrms_angle_Free'] = ptrms_angle
def test_get_dec_and_inc(self): # testing full thing with real data
dec, inc, intenstiy, tau, V, means, PCA_sigma = lib_direct.get_dec_and_inc(PintPars_example.zdata, PintPars_example.t_Arai, PintPars_example.tmin, PintPars_example.tmax, anchored=False)
self.assertAlmostEqual(dec, 267.4620127216387)
self.assertAlmostEqual(inc, 86.349431762792364)
self.assertGreaterEqual(tau[0], tau[1])
self.assertGreaterEqual(tau[1], tau[2])
