def test_dep_par_jet(plot=False): import numpy as np from jetset.jet_model import Jet j = Jet(emitters_distribution='plc') j.add_user_par(name='B0',units='G',val=1E-5,val_min=0,val_max=None) j.make_dependent_par(par='B',depends_on=['B0'],par_expr='B0*5') np.testing.assert_allclose(j.parameters.B.val, j.parameters.B0.val*5) j.save_model('test.pkl') new_j=Jet.load_model('test.pkl') new_j.parameters.B0.val=1 np.testing.assert_allclose(new_j.parameters.B.val, new_j.parameters.B0.val*5)
def test_custom_emitters(plot=True): from jetset.jet_model import Jet from jetset.jet_emitters import EmittersDistribution import numpy as np def distr_func_bkn(gamma_break, gamma, s1, s2): return np.power(gamma, -s1) * (1. + (gamma / gamma_break))**(-(s2 - s1)) n_e = EmittersDistribution('custom_bkn', spectral_type='bkn') n_e.add_par('gamma_break', par_type='turn-over-energy', val=1E3, vmin=1., vmax=None, unit='lorentz-factor') n_e.add_par('s1', par_type='LE_spectral_slope', val=2.5, vmin=-10., vmax=10, unit='') n_e.add_par('s2', par_type='LE_spectral_slope', val=3.2, vmin=-10., vmax=10, unit='') n_e.set_distr_func(distr_func_bkn) n_e.parameters.show_pars() n_e.parameters.s1.val = 2.0 n_e.parameters.s2.val = 3.5 if plot is True: n_e.plot() my_jet = Jet(emitters_distribution=n_e) my_jet.Norm_distr = True my_jet.parameters.N.val = 5E4 my_jet.eval() np.testing.assert_allclose(my_jet.emitters_distribution.eval_N(), my_jet.parameters.N.val, rtol=1E-5) print(my_jet.emitters_distribution.eval_N(), my_jet.parameters.N.val) print(n_e.eval_N(), my_jet.parameters.N.val) assert (my_jet.emitters_distribution.emitters_type == 'electrons') my_jet.save_model('test_jet_custom_emitters.pkl') my_jet = Jet.load_model('test_jet_custom_emitters.pkl') my_jet.eval()
def test_jet(plot=True): #print('--------> plot',plot) from jetset.jet_model import Jet j = Jet() j.eval() j.energetic_report() if plot is True: j.plot_model() j.emitters_distribution.plot() j.emitters_distribution.plot2p() j.emitters_distribution.plot3p() j.emitters_distribution.plot3p(energy_unit='eV') j.emitters_distribution.plot3p(energy_unit='erg') j.save_model('test_jet.pkl') j_new = Jet.load_model('test_jet.pkl')
def test_custom_emitters_array(plot=True): from jetset.jet_model import Jet from jetset.jet_emitters import EmittersArrayDistribution import numpy as np # gamma array gamma = np.logspace(1, 8, 500) # gamma array this is n(\gamma) in 1/cm^3/gamma n_gamma = gamma**-2 * 1E-5 * np.exp(-gamma / 1E5) N1 = np.trapz(n_gamma, gamma) n_distr = EmittersArrayDistribution(name='array_distr', emitters_type='protons', gamma_array=gamma, n_gamma_array=n_gamma, normalize=False) N2 = np.trapz(n_distr._array_n_gamma, n_distr._array_gamma) j = Jet(emitters_distribution=n_distr, verbose=False) j.parameters.z_cosm.val = z = 0.001 j.parameters.beam_obj.val = 1 j.parameters.N.val = 1 j.parameters.NH_pp.val = 1 j.parameters.B.val = 0.01 j.parameters.R.val = 1E18 j.set_IC_nu_size(100) j.gamma_grid_size = 200 N3 = np.trapz(j.emitters_distribution.n_gamma_p, j.emitters_distribution.gamma_p) np.testing.assert_allclose(N1, N2, rtol=1E-5) np.testing.assert_allclose(N1, N3, rtol=1E-2) np.testing.assert_allclose(N1, j.emitters_distribution.eval_N(), rtol=1E-2) assert (j.emitters_distribution.emitters_type == 'protons') j.eval() j.save_model('test_jet_custom_emitters_array.pkl') j = Jet.load_model('test_jet_custom_emitters_array.pkl') j.eval()
def test_jet(plot=True): print('--------> test_jet', plot) from jetset.jet_model import Jet j = Jet() j.eval() j.energetic_report() sum1 = j.spectral_components.Sum.SED.nuFnu if plot is True: j.plot_model() j.emitters_distribution.plot() j.emitters_distribution.plot2p() j.emitters_distribution.plot3p() j.emitters_distribution.plot3p(energy_unit='eV') j.emitters_distribution.plot3p(energy_unit='erg') j.save_model('test_jet.pkl') j_new = Jet.load_model('test_jet.pkl') j_new.eval() sum2 = j_new.spectral_components.Sum.SED.nuFnu np.testing.assert_allclose(sum2, sum1, rtol=1E-5)
def test_hadronic_jet(plot=True): from jetset.jet_model import Jet j = Jet(proton_distribution='plc') j.parameters.gmin.val = 2 j.parameters.gmax.val = 1E8 j.parameters.NH_pp.val = 1E10 j.parameters.N.val = 1E1 j.parameters.B.val = 80 j.parameters.p.val = 2.5 j.eval() j.show_model() sum1 = j.spectral_components.Sum.SED.nuFnu if plot is True: j.plot_model() j.save_model('test_jet_hadronic.pkl') j_new = Jet.load_model('test_jet_hadronic.pkl') j_new.eval() sum2 = j_new.spectral_components.Sum.SED.nuFnu np.testing.assert_allclose(sum2, sum1, rtol=1E-5)
def test_dep_par(plot=False): from jetset.jet_emitters import EmittersDistribution import numpy as np def distr_func_bkn(gamma_break, gamma, s1, s2): return np.power(gamma, -s1) * (1. + (gamma / gamma_break)) ** (-(s2 - s1)) n_e_bkn = EmittersDistribution('bkn', spectral_type='bkn') n_e_bkn.add_par('gamma_break', par_type='turn-over-energy', val=1E3, vmin=1., vmax=None, unit='lorentz-factor') n_e_bkn.add_par('s1', par_type='LE_spectral_slope', val=2.5, vmin=-10., vmax=10, unit='') n_e_bkn.add_par('s2', par_type='LE_spectral_slope', val=3.2, vmin=-10., vmax=10, unit='') n_e_bkn.set_distr_func(distr_func_bkn) n_e_bkn.parameters.show_pars() n_e_bkn.parameters.s1.val = 2.0 n_e_bkn.parameters.s2.val = 3.5 n_e_bkn.update() n_e_bkn.parameters.show_pars() from jetset.jet_model import Jet j = Jet(emitters_distribution=n_e_bkn) # def par_func(s1): # return s1+1 j.make_dependent_par(par='s2', depends_on=['s1'], par_expr='s1+1') print('here') j.parameters.s1.val = 3 print('done') np.testing.assert_allclose(j.parameters.s2.val, j.parameters.s1.val + 1) j.save_model('jet.pkl') new_jet = Jet.load_model('jet.pkl') print('here') new_jet.parameters.s1.val = 2 print('done') np.testing.assert_allclose(new_jet.parameters.s2.val, new_jet.parameters.s1.val + 1) j.eval() new_jet.show_model()