def test_image_plot_ratio_callable():
"""
Test plot_pull by comparing against a reference image generated via
`pytest --mpl-generate-path=tests/baseline`
"""
np.random.seed(42)
hist_1 = Hist(
axis.Regular(50,
-5,
5,
name="X",
label="x [units]",
underflow=False,
overflow=False)).fill(np.random.normal(size=1000))
def model(x, a=1 / np.sqrt(2 * np.pi), x0=0, sigma=1, offset=0):
return a * np.exp(-((x - x0)**2) / (2 * sigma**2)) + offset
fig = plt.figure()
assert hist_1.plot_ratio(model,
eb_color="black",
fp_color="blue",
ub_color="lightblue")
return fig
def test_image_plot_ratio_hist():
"""
Test plot_pull by comparing against a reference image generated via
`pytest --mpl-generate-path=tests/baseline`
"""
np.random.seed(42)
hist_1 = Hist(
axis.Regular(50,
-5,
5,
name="X",
label="x [units]",
underflow=False,
overflow=False)).fill(np.random.normal(size=1000))
hist_2 = Hist(
axis.Regular(50,
-5,
5,
name="X",
label="x [units]",
underflow=False,
overflow=False)).fill(np.random.normal(size=1700))
fig = plt.figure()
assert hist_1.plot_ratio(hist_2,
rp_num_label="numerator",
rp_denom_label="denominator")
return fig