1 files changed, 39 insertions, 1 deletions
diff --git a/test/test_gf_simple.py b/test/test_gf_simple.py
index 51460c1..3a6ebb5 100644
--- a/test/test_gf_simple.py
+++ b/test/test_gf_simple.py
@@ -1,11 +1,20 @@
+import numpy as np
+import matplotlib as mpl
+mpl.use('Agg')
+import matplotlib.pyplot as plt
+
 import GolemFitPy as gf
 
+FASTMODE = True
+
 dp = gf.DataPaths()
 npp = gf.NewPhysicsParams()
 sp = gf.SteeringParams(gf.sampleTag.HESE)
 
 sp.quiet = False
-# sp.fastmode = True
+if FASTMODE:
+    sp.fastmode = True
+sp.frequentist = True
 
 golem = gf.GolemFit(dp, sp, npp)
 
@@ -20,4 +29,33 @@ fp_sh.astroFlavorAngle1 = 0.36
 fp_sh.astroFlavorAngle2 = -0.57
 
 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')