def test_atmo(arr):
x = simple_atmo(arr, 0.03, 10, 0.15)
assert x[0][0][0] - 0.080560341 < 1e-4
# Gamma output is not within the range 0..1
with pytest.raises(ValueError):
x = simple_atmo(arr, 2.0, 10, 0.15)
# Sigmoidal contrast output contains NaN
with pytest.raises(ValueError):
x = simple_atmo(arr, 0.03, 1000, -0.15)
def test_simple_atmos_opstring(arr):
x = simple_atmo(arr, 0.03, 10, 0.15)
ops = simple_atmo_opstring(0.03, 10, 0.15)
for op in parse_operations(ops):
arr = op(arr)
assert np.allclose(x, arr)
