diff options
Diffstat (limited to 'test/test_gf_simple.py')
| -rw-r--r-- | test/test_gf_simple.py | 66 |
1 files changed, 0 insertions, 66 deletions
diff --git a/test/test_gf_simple.py b/test/test_gf_simple.py deleted file mode 100644 index daa14a4..0000000 --- a/test/test_gf_simple.py +++ /dev/null @@ -1,66 +0,0 @@ -import numpy as np -import matplotlib as mpl -mpl.use('Agg') -import matplotlib.pyplot as plt - -import GolemFitPy as gf - -FASTMODE = False - -dp = gf.DataPaths() -npp = gf.NewPhysicsParams() -sp = gf.SteeringParams(gf.sampleTag.MagicTau) - -sp.quiet = False -if FASTMODE: - sp.fastmode = True -# sp.frequentist = True -sp.load_data_from_text_file = False - -golem = gf.GolemFit(dp, sp, npp) - -fp = gf.FitParameters(gf.sampleTag.MagicTau) -fp.astroFlavorAngle1 = 4./9. -fp.astroFlavorAngle2 = 0. - -# golem.SetupAsimov(fp) -seed = 0 -golem.Swallow(golem.SpitRealization(fp, seed)) - -fp_sh = gf.FitParameters(gf.sampleTag.MagicTau) -# fp_sh.astroFlavorAngle1 = 0.36 -# fp_sh.astroFlavorAngle2 = -0.57 -fp_sh.astroFlavorAngle1 = 0. -fp_sh.astroFlavorAngle2 = 1. - -print 'Eval fp = {0}'.format(golem.EvalLLH(fp)) - -# energy_centers = golem.GetEnergyBinsMC()[:-1]+ np.diff(golem.GetEnergyBinsMC())/2. - -# plt.hist(energy_centers,bins=golem.GetEnergyBinsMC(), -# weights=np.sum(golem.GetExpectation(fp),axis=(0,1,2,3)), -# histtype="step", lw = 2, label='injected') - -# data_energy_dist = np.sum(golem.GetDataDistribution(),axis=(0,1,2,3)) -# energy_centers=golem.GetEnergyBinsData()[:-1]+ np.diff(golem.GetEnergyBinsData())/2. -# plt.errorbar(energy_centers,data_energy_dist,yerr = np.sqrt(data_energy_dist),fmt='o') - -print 'Eval fp_sh = {0}'.format(golem.EvalLLH(fp_sh)) - -# plt.hist(energy_centers,bins=golem.GetEnergyBinsMC(), -# weights=np.sum(golem.GetExpectation(fp_sh),axis=(0,1,2,3)), -# histtype="step", lw = 2, label='test') - -# data_energy_dist = np.sum(golem.GetDataDistribution(),axis=(0,1,2,3)) -# energy_centers=golem.GetEnergyBinsData()[:-1]+ np.diff(golem.GetEnergyBinsData())/2. -# plt.errorbar(energy_centers,data_energy_dist,yerr = np.sqrt(data_energy_dist),fmt='o') - -# plt.loglog(nonposy="clip") -# plt.xlabel(r"Deposited energy/GeV") -# plt.ylabel(r"Events") - -# outname = 'Expectation' -# if FASTMODE: -# plt.savefig(outname + 'fastmode.png') -# else: -# plt.savefig(outname + '.png') |
