def test_imcompatible_areas(self, sunz_ds2, sunz_sza):
"""Test sunz correction on incompatible areas."""
from satpy.composites import IncompatibleAreas
from satpy.modifiers.geometry import SunZenithCorrector
comp = SunZenithCorrector(name='sza_test', modifiers=tuple(), correction_limit=90)
with pytest.raises(IncompatibleAreas):
comp((sunz_ds2, sunz_sza), test_attr='test')
def test_basic_default_provided(self, data_arr, sunz_sza):
"""Test default limits when SZA is provided."""
from satpy.modifiers.geometry import SunZenithCorrector
comp = SunZenithCorrector(name='sza_test', modifiers=tuple())
res = comp((data_arr, sunz_sza), test_attr='test')
np.testing.assert_allclose(
res.values,
np.array([[22.401667, 22.31777], [22.437503, 22.353533]]))
def test_basic_lims_provided(self, data_arr, sunz_sza):
"""Test custom limits when SZA is provided."""
from satpy.modifiers.geometry import SunZenithCorrector
comp = SunZenithCorrector(name='sza_test',
modifiers=tuple(),
correction_limit=90)
res = comp((data_arr, sunz_sza), test_attr='test')
np.testing.assert_allclose(
res.values,
np.array([[66.853262, 68.168939], [66.30742, 67.601493]]))
def test_basic_default_not_provided(self, sunz_ds1):
"""Test default limits when SZA isn't provided."""
from satpy.modifiers.geometry import SunZenithCorrector
comp = SunZenithCorrector(name='sza_test', modifiers=tuple())
res = comp((sunz_ds1,), test_attr='test')
np.testing.assert_allclose(res.values, np.array([[22.401667, 22.31777], [22.437503, 22.353533]]))
assert 'y' in res.coords
assert 'x' in res.coords
ds1 = sunz_ds1.copy().drop_vars(('y', 'x'))
res = comp((ds1,), test_attr='test')
np.testing.assert_allclose(res.values, np.array([[22.401667, 22.31777], [22.437503, 22.353533]]))
assert 'y' not in res.coords
assert 'x' not in res.coords