def test_BinnedChi2(self):
f = gaussian
assert_equal(list(describe(f)), ['x', 'mean', 'sigma'])
lh = BinnedChi2(gaussian, self.data, bound=[-3, 3])
assert_equal(list(describe(lh)), ['mean', 'sigma'])
assert_almost_equal(lh(0, 1), 19951.005399882044, 1)
minuit = iminuit.Minuit(lh)
assert_equal(minuit.errordef, 1.0)
def test_BinnedChi2(self):
f = gaussian
assert list(describe(f)) == ['x', 'mean', 'sigma']
lh = BinnedChi2(gaussian, self.data, bound=[-3, 3])
assert list(describe(lh)) == ['mean', 'sigma']
assert_allclose(lh(0, 1), 19951.005399882044, atol=1)
minuit = iminuit.Minuit(lh)
assert_allclose(minuit.errordef, 1.0)
def test_bx2_with_parts():
np.random.seed(0)
data = np.random.randn(10000)
shifted = data + 3.
data = np.append(data, [shifted])
g1 = Extended(rename(gaussian, ['x', 'lmu', 'lsigma']), extname='N1')
g2 = Extended(rename(gaussian, ['x', 'rmu', 'rsigma']), extname='N2')
allpdf = AddPdf(g1, g2)
bx2 = BinnedChi2(allpdf, data)
bx2.draw(args=(0, 1, 10000, 3, 1, 10000), parts=True)
def test_bx2_with_parts():
np.random.seed(0)
data = np.random.randn(10000)
shifted = data + 3.0
data = np.append(data, [shifted])
g1 = Extended(rename(gaussian, ["x", "lmu", "lsigma"]), extname="N1")
g2 = Extended(rename(gaussian, ["x", "rmu", "rsigma"]), extname="N2")
allpdf = AddPdf(g1, g2)
bx2 = BinnedChi2(allpdf, data)
bx2.draw(args=(0, 1, 10000, 3, 1, 10000), parts=True)
def test_draw_bx2():
np.random.seed(0)
data = np.random.randn(1000)
blh = BinnedChi2(Extended(gaussian), data)
blh.draw(args=(0., 1., 1000))