def test_stat_surface_reoptimize():
dataset = MyDataset()
fit = Fit()
fit.run([dataset])
dataset.models.parameters["z"].value = 0
x_values = [1, 2, 3]
y_values = [2e2, 3e2, 4e2]
result = fit.stat_surface(datasets=[dataset],
x="x",
y="y",
x_values=x_values,
y_values=y_values,
reoptimize=True)
assert_allclose(result["x_scan"], x_values, atol=1e-7)
assert_allclose(result["y_scan"], y_values, atol=1e-7)
expected_stat = [
[1.0001e04, 1.0000e00, 1.0001e04],
[1.0000e04, 0.0000e00, 1.0000e04],
[1.0001e04, 1.0000e00, 1.0001e04],
]
assert_allclose(list(result["stat_scan"]), expected_stat, atol=1e-7)
assert_allclose(result["fit_results"][0][0].total_stat,
result["stat_scan"][0][0],
atol=1e-7)
def test_stat_surface():
dataset = MyDataset()
fit = Fit()
fit.run([dataset])
x_values = [1, 2, 3]
y_values = [2e2, 3e2, 4e2]
result = fit.stat_surface(datasets=[dataset],
x="x",
y="y",
x_values=x_values,
y_values=y_values)
assert_allclose(result["x_scan"], x_values, atol=1e-7)
assert_allclose(result["y_scan"], y_values, atol=1e-7)
expected_stat = [
[1.0001e04, 1.0000e00, 1.0001e04],
[1.0000e04, 0.0000e00, 1.0000e04],
[1.0001e04, 1.0000e00, 1.0001e04],
]
assert_allclose(list(result["stat_scan"]), expected_stat, atol=1e-7)
assert len(result["fit_results"]) == 0
# Check that original value state wasn't changed
assert_allclose(dataset.models.parameters["x"].value, 2)
assert_allclose(dataset.models.parameters["y"].value, 3e2)
