def test_scatter(flat, orders, settings):
# The background scatter step in reduce possibly uses a
# different set of files for the image
# However it should still be able to create a scatter fit
# from the flat image as is done here
img, _ = flat
orders, column_range = orders
settings = settings["scatter"]
settings["sigma_cutoff"] = settings["scatter_cutoff"]
del settings["scatter_cutoff"]
if img is None:
pytest.skip("Need flat")
scatter = estimate_background_scatter(img,
orders,
column_range=column_range,
**settings)
degree = settings["scatter_degree"]
if np.isscalar(degree):
degree = [degree, degree]
assert isinstance(scatter, np.ndarray)
assert scatter.ndim == 2
assert scatter.shape[0] == degree[0] + 1
assert scatter.shape[1] == degree[1] + 1
def test_scatter(flat, orders):
img, _ = flat
orders, column_range = orders
scatter = estimate_background_scatter(img,
orders,
scatter_degree=4,
column_range=column_range)
assert isinstance(scatter, np.ndarray)
assert scatter.ndim == 2
assert scatter.shape[0] == 5
assert scatter.shape[1] == 5
def test_simple():
img = np.full((100, 100), 10.0)
orders = np.array([[0, 25], [0, 50], [0, 75]])
scatter = estimate_background_scatter(img,
orders,
scatter_degree=0,
plot=False)
assert isinstance(scatter, np.ndarray)
assert scatter.ndim == 2
assert scatter.shape[0] == 1
assert scatter.shape[1] == 1
assert np.allclose(scatter[0, 0], 10.0)
def test_scatter_degree():
img = np.full((100, 100), 10.0)
orders = np.full((2, 2), 1.0)
estimate_background_scatter(img, orders, scatter_degree=0)
with pytest.raises(ValueError):
estimate_background_scatter(img, orders, scatter_degree=-1)
estimate_background_scatter(img, orders, scatter_degree=(2, 2))
with pytest.raises(AssertionError):
estimate_background_scatter(img, orders, scatter_degree=(1, ))
with pytest.raises(AssertionError):
estimate_background_scatter(img, orders, scatter_degree=(3, 2, 1))
with pytest.raises(ValueError):
estimate_background_scatter(img, orders, scatter_degree=(2, -1))