def test_download_from_zenodo(self):
"""Test downloading SEVIRI HRIT data from zenodo."""
from satpy.demo import download_seviri_hrit_20180228_1500
self.get_mock.side_effect = _FakeRequest
with mock_filesystem():
download_seviri_hrit_20180228_1500()
assert _FakeRequest.requests_log[0].startswith("https://zenodo.org")
def setup_cache(self):
"""Fetch the data files."""
try:
from satpy.demo import download_seviri_hrit_20180228_1500
download_seviri_hrit_20180228_1500()
except ImportError:
assert len(get_filenames(self.subdir)) == 114
download_rsr()
download_luts(aerosol_type='rayleigh_only')
def test_do_not_download_same_file_twice(self):
"""Test that files are not downloaded twice."""
from satpy.demo import download_seviri_hrit_20180228_1500
get_mock = mock.MagicMock()
self.get_mock.return_value.__enter__ = get_mock
with mock_filesystem():
files = download_seviri_hrit_20180228_1500(subset={"HRV": [1, 2, 3], "IR_108": [1, 2], "EPI": None})
new_files = download_seviri_hrit_20180228_1500(subset={"HRV": [1, 2, 3], "IR_108": [1, 2], "EPI": None})
assert set(files) == set(new_files)
assert get_mock.call_count == 6
def test_download_to_output_directory(self):
"""Test downloading to an output directory."""
from satpy.demo import download_seviri_hrit_20180228_1500
from tempfile import gettempdir
with mock_filesystem():
base_dir = gettempdir()
files = download_seviri_hrit_20180228_1500(base_dir=base_dir)
assert files[0].startswith(base_dir)
def test_download_gets_files_with_contents(self):
"""Test downloading SEVIRI HRIT data with content."""
from satpy.demo import download_seviri_hrit_20180228_1500
self.get_mock.side_effect = _FakeRequest
with mock_filesystem():
files = download_seviri_hrit_20180228_1500()
assert len(files) == 114
assert set(files) == set(self.files)
for the_file in files:
with open(the_file, mode="r") as fd:
assert fd.read().decode("utf8") == os.path.basename(the_file)
def test_download_a_subset_of_files(self):
"""Test downloading a subset of files."""
from satpy.demo import download_seviri_hrit_20180228_1500
with mock_filesystem():
files = download_seviri_hrit_20180228_1500(subset={"HRV": [1, 2, 3], "IR_108": [1, 2], "EPI": None})
assert set(files) == set(os.path.join(self.subdir, filename) for filename in [
'H-000-MSG4__-MSG4________-_________-EPI______-201802281500-__',
'H-000-MSG4__-MSG4________-HRV______-000001___-201802281500-__',
'H-000-MSG4__-MSG4________-HRV______-000002___-201802281500-__',
'H-000-MSG4__-MSG4________-HRV______-000003___-201802281500-__',
'H-000-MSG4__-MSG4________-IR_108___-000001___-201802281500-__',
'H-000-MSG4__-MSG4________-IR_108___-000002___-201802281500-__',
])