aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_gf.py
diff options
context:
space:
mode:
authorShivesh Mandalia <shivesh.mandalia@outlook.com>2020-02-29 02:18:50 +0000
committerShivesh Mandalia <shivesh.mandalia@outlook.com>2020-02-29 02:18:50 +0000
commitb337b7a457341999f97a188945c2c4cc03f7b11c (patch)
tree820f45be852f94ae68fb4a407d677345366db02b /test/test_gf.py
parent7b32b3e2c437f65f6ac946d16463691e7496be29 (diff)
downloadGolemFlavor-b337b7a457341999f97a188945c2c4cc03f7b11c.tar.gz
GolemFlavor-b337b7a457341999f97a188945c2c4cc03f7b11c.zip
move golemfit test to another repo and slightly reluctantly use american style flavor spelling consistently
Diffstat (limited to 'test/test_gf.py')
-rw-r--r--test/test_gf.py89
1 files changed, 0 insertions, 89 deletions
diff --git a/test/test_gf.py b/test/test_gf.py
deleted file mode 100644
index 4b77924..0000000
--- a/test/test_gf.py
+++ /dev/null
@@ -1,89 +0,0 @@
-import numpy.ma as ma
-
-import GolemFitPy as gf
-
-FASTMODE = True
-
-def steering_params():
- steering_categ = 'p2_0'
- params = gf.SteeringParams(gf.sampleTag.HESE)
- if FASTMODE:
- params.fastmode = True
- params.quiet = False
- params.simToLoad= steering_categ.lower()
- return params
-
-def set_up_as(fitter, params):
- print 'Injecting the model', params
- asimov_params = gf.FitParameters(gf.sampleTag.HESE)
- for x in params.iterkeys():
- asimov_params.__setattr__(x, float(params[x]))
- fitter.SetupAsimov(asimov_params)
-
-def setup_fitter(asimov_paramset):
- datapaths = gf.DataPaths()
- sparams = steering_params()
- npp = gf.NewPhysicsParams()
- fitter = gf.GolemFit(datapaths, sparams, npp)
- set_up_as(fitter, asimov_paramset)
- return fitter
-
-def get_llh(fitter, params):
- fitparams = gf.FitParameters(gf.sampleTag.HESE)
- for x in params.iterkeys():
- fitparams.__setattr__(x, float(params[x]))
- llh = -fitter.EvalLLH(fitparams)
- return llh
-
-asimov_paramset = {'astroFlavorAngle1': 4/9., 'astroFlavorAngle2': 0.}
-print 'injecting', asimov_paramset
-fitter = setup_fitter(asimov_paramset)
-
-test_paramset = {'astroFlavorAngle1': 0.36, 'astroFlavorAngle2': -0.57}
-print 'testing', test_paramset
-print 'llh', get_llh(fitter, test_paramset)
-
-test_paramset = {'astroFlavorAngle1': 0.385224559219, 'astroFlavorAngle2': -0.157617854374}
-print 'testing', test_paramset
-print 'llh', get_llh(fitter, test_paramset)
-
-test_paramset = {'astroFlavorAngle1': 0.415578500878, 'astroFlavorAngle2': -0.0196186993217}
-print 'testing', test_paramset
-print 'llh', get_llh(fitter, test_paramset)
-
-test_paramset = {'astroFlavorAngle1': 4/9., 'astroFlavorAngle2': 0}
-print 'testing', test_paramset
-print 'llh', get_llh(fitter, test_paramset)
-
-test_paramset = {'astroFlavorAngle1': 4/9., 'astroFlavorAngle2': 0}
-print 'testing', test_paramset
-print 'llh', get_llh(fitter, test_paramset)
-
-
-import numpy as np
-import matplotlib as mpl
-mpl.use('Agg')
-import matplotlib.pyplot as plt
-
-shape = (10, 100)
-angle1_binning = np.linspace(0, 1, shape[0])
-angle2_binning = np.linspace(-1, 1, shape[1])
-
-for an1 in angle1_binning:
- llhs = []
- for an2 in angle2_binning:
- test_paramset = {'astroFlavorAngle1': an1, 'astroFlavorAngle2': an2}
- llh = get_llh(fitter, test_paramset)
- if abs(llh) > 9e9:
- llhs.append(np.nan)
- else:
- llhs.append(llh)
- llhs = ma.masked_invalid(llhs)
- plt.plot(angle2_binning, llhs, label='astroFlavorAngle1 = {0}'.format(an1))
-plt.xlabel('astroFlavorAngle2')
-plt.ylabel('LLH')
-plt.legend()
-if FASTMODE:
- plt.savefig('llh_profile_fastmode.png'.format(an1))
-else:
- plt.savefig('llh_profile.png'.format(an1))