Skip to content

Commit

Permalink
Refactor tests
Browse files Browse the repository at this point in the history
  • Loading branch information
mraspaud committed Jul 20, 2021
1 parent 1457f8b commit b6e02e1
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions satpy/tests/reader_tests/test_olci_nc.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,27 +243,25 @@ def test_olci_angles_caches_interpolation(self):
from satpy.tests.utils import make_dataid

ds_id = make_dataid(name='solar_zenith_angle')
with mock.patch("geotiepoints.interpolator.Interpolator") as interpolator:
interpolator.return_value.interpolate.return_value = (
self.expected_data, self.expected_data, self.expected_data)

self.file_handler.get_dataset(ds_id, self.filename_info)
self.file_handler.get_dataset(ds_id, self.filename_info)
assert(interpolator.call_count == 1)
self._check_interpolator_is_called_only_once(ds_id, ds_id)

def test_olci_different_angles_caches_interpolation(self):
"""Test reading different angles datasets caches interpolation."""
from satpy.tests.utils import make_dataid

ds_id_zenith = make_dataid(name='solar_zenith_angle')
ds_id_azimuth = make_dataid(name='solar_azimuth_angle')
self._check_interpolator_is_called_only_once(ds_id_azimuth, ds_id_zenith)

def _check_interpolator_is_called_only_once(self, ds_id_1, ds_id_2):
"""Check that the interpolation is used only once."""
with mock.patch("geotiepoints.interpolator.Interpolator") as interpolator:
interpolator.return_value.interpolate.return_value = (
self.expected_data, self.expected_data, self.expected_data)

self.file_handler.get_dataset(ds_id_zenith, self.filename_info)
self.file_handler.get_dataset(ds_id_azimuth, self.filename_info)
assert(interpolator.call_count == 1)
self.file_handler.get_dataset(ds_id_2, self.filename_info)
self.file_handler.get_dataset(ds_id_1, self.filename_info)
assert (interpolator.call_count == 1)

def tearDown(self):
"""Tear down the test case."""
Expand Down

0 comments on commit b6e02e1

Please sign in to comment.