Source code for satpy.tests.reader_tests.test_viirs_sdr
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2017-2019, 2022, 2023 Satpy developers
#
# This file is part of satpy.
#
# satpy is free software: you can redistribute it and/or modify it under the
# terms of the GNU General Public License as published by the Free Software
# Foundation, either version 3 of the License, or (at your option) any later
# version.
#
# satpy is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# satpy. If not, see <http://www.gnu.org/licenses/>.
"""Module for testing the satpy.readers.viirs_sdr module."""
import os
import unittest
from contextlib import contextmanager
from unittest import mock
import numpy as np
import pytest
from satpy.readers.viirs_atms_sdr_base import DATASET_KEYS
from satpy.tests.reader_tests.test_hdf5_utils import FakeHDF5FileHandler
DEFAULT_FILE_DTYPE = np.uint16
DEFAULT_FILE_SHAPE = (32, 300)
# Mimicking one scan line of data
DEFAULT_FILE_DATA = np.arange(DEFAULT_FILE_SHAPE[0] * DEFAULT_FILE_SHAPE[1],
dtype=DEFAULT_FILE_DTYPE).reshape(DEFAULT_FILE_SHAPE)
DEFAULT_FILE_FACTORS = np.array([2.0, 1.0], dtype=np.float32)
[docs]
class FakeHDF5FileHandler2(FakeHDF5FileHandler):
"""Swap-in HDF5 File Handler."""
_num_test_granules = 1
_num_scans_per_gran = [48]
def __init__(self, filename, filename_info, filetype_info, include_factors=True):
"""Create fake file handler."""
self.include_factors = include_factors
super(FakeHDF5FileHandler2, self).__init__(filename, filename_info, filetype_info)
[docs]
@staticmethod
def _add_basic_metadata_to_file_content(file_content, filename_info, num_grans):
start_time = filename_info["start_time"]
end_time = filename_info["end_time"].replace(year=start_time.year,
month=start_time.month,
day=start_time.day)
begin_date = start_time.strftime("%Y%m%d")
begin_date = np.array(begin_date)
begin_time = start_time.strftime("%H%M%S.%fZ")
begin_time = np.array(begin_time)
ending_date = end_time.strftime("%Y%m%d")
ending_time = end_time.strftime("%H%M%S.%fZ")
new_file_content = {
"{prefix2}/attr/AggregateNumberGranules": num_grans,
"{prefix2}/attr/AggregateBeginningDate": begin_date,
"{prefix2}/attr/AggregateBeginningTime": begin_time,
"{prefix2}/attr/AggregateEndingDate": ending_date,
"{prefix2}/attr/AggregateEndingTime": ending_time,
"{prefix2}/attr/G-Ring_Longitude": np.array([0.0, 0.1, 0.2, 0.3]),
"{prefix2}/attr/G-Ring_Latitude": np.array([0.0, 0.1, 0.2, 0.3]),
"{prefix2}/attr/AggregateBeginningOrbitNumber": "{0:d}".format(filename_info["orbit"]),
"{prefix2}/attr/AggregateEndingOrbitNumber": "{0:d}".format(filename_info["orbit"]),
"{prefix1}/attr/Instrument_Short_Name": "VIIRS",
"/attr/Platform_Short_Name": "NPP",
}
file_content.update(new_file_content)
[docs]
def _add_granule_specific_info_to_file_content(
self,
file_content, dataset_group, num_granules, num_scans_per_granule,
gran_group_prefix):
lons_lists = self._get_per_granule_lons()
lats_lists = self._get_per_granule_lats()
file_content["{prefix3}/NumberOfScans"] = np.array([48] * num_granules)
for granule_idx in range(num_granules):
prefix_gran = "{prefix}/{dataset_group}_Gran_{idx}".format(prefix=gran_group_prefix,
dataset_group=dataset_group,
idx=granule_idx)
num_scans = num_scans_per_granule[granule_idx]
file_content[prefix_gran + "/attr/N_Number_Of_Scans"] = num_scans
file_content[prefix_gran + "/attr/G-Ring_Longitude"] = lons_lists[granule_idx]
file_content[prefix_gran + "/attr/G-Ring_Latitude"] = lats_lists[granule_idx]
[docs]
@staticmethod
def _get_per_granule_lons():
return [
np.array(
[
50.51393, 49.566296, 48.865967, 18.96082, -4.0238385,
-7.05221, -10.405702, 14.638646
],
dtype=np.float32),
np.array(
[
53.52594, 51.685738, 50.439102, 14.629087, -10.247547,
-13.951393, -18.256989, 8.36572
],
dtype=np.float32),
np.array(
[
59.386833, 55.770416, 53.38952, 8.353765, -18.062435,
-22.608992, -27.867302, -1.3537619
],
dtype=np.float32),
np.array(
[
72.50243, 64.17125, 59.15234, -1.3654504, -27.620953,
-33.091743, -39.28113, -17.749891
],
dtype=np.float32)
]
[docs]
@staticmethod
def _get_per_granule_lats():
return [
np.array(
[
67.969505, 65.545685, 63.103046, 61.853905, 55.169273,
57.062447, 58.86063, 66.495514
],
dtype=np.float32),
np.array(
[
72.74879, 70.2493, 67.84738, 66.49691, 58.77254,
60.465942, 62.11525, 71.08249
],
dtype=np.float32),
np.array(
[
77.393425, 74.977875, 72.62976, 71.083435, 62.036346,
63.465122, 64.78075, 75.36842
],
dtype=np.float32),
np.array(
[
81.67615, 79.49934, 77.278656, 75.369415, 64.72178,
65.78417, 66.66166, 79.00025
],
dtype=np.float32),
]
[docs]
def _add_data_info_to_file_content(self, file_content, filename, data_var_prefix, num_grans):
# SDR files always produce data with 48 scans per granule even if there are less
total_rows = DEFAULT_FILE_SHAPE[0] * 48 * num_grans
new_shape = (total_rows, DEFAULT_FILE_SHAPE[1])
if filename[2:5] in ["M{:02d}".format(x) for x in range(12)] + ["I01", "I02", "I03"]:
keys = ["Radiance", "Reflectance"]
elif filename[2:5] in ["M{:02d}".format(x) for x in range(12, 17)] + ["I04", "I05"]:
keys = ["Radiance", "BrightnessTemperature"]
else:
# DNB
keys = ["Radiance"]
for k in keys:
k = data_var_prefix + "/" + k
file_content[k] = np.repeat(DEFAULT_FILE_DATA.copy(), 48 * num_grans, axis=0)
file_content[k + "/shape"] = new_shape
if self.include_factors:
file_content[k + "Factors"] = np.repeat(
DEFAULT_FILE_FACTORS.copy()[None, :], num_grans, axis=0).ravel()
[docs]
@staticmethod
def _add_geolocation_info_to_file_content(file_content, filename, data_var_prefix, num_grans):
# SDR files always produce data with 48 scans per granule even if there are less
total_rows = DEFAULT_FILE_SHAPE[0] * 48 * num_grans
new_shape = (total_rows, DEFAULT_FILE_SHAPE[1])
is_dnb = filename[:5] not in ["GMODO", "GIMGO"]
if not is_dnb:
lon_data = np.linspace(15, 55, DEFAULT_FILE_SHAPE[1]).astype(DEFAULT_FILE_DTYPE)
lat_data = np.linspace(55, 75, DEFAULT_FILE_SHAPE[1]).astype(DEFAULT_FILE_DTYPE)
else:
lon_data = np.linspace(5, 45, DEFAULT_FILE_SHAPE[1]).astype(DEFAULT_FILE_DTYPE)
lat_data = np.linspace(45, 65, DEFAULT_FILE_SHAPE[1]).astype(DEFAULT_FILE_DTYPE)
for k in ["Latitude"]:
k = data_var_prefix + "/" + k
file_content[k] = lat_data
file_content[k] = np.repeat([file_content[k]], total_rows, axis=0)
file_content[k + "/shape"] = new_shape
for k in ["Longitude"]:
k = data_var_prefix + "/" + k
file_content[k] = lon_data
file_content[k] = np.repeat([file_content[k]], total_rows, axis=0)
file_content[k + "/shape"] = new_shape
angles = ["SolarZenithAngle",
"SolarAzimuthAngle",
"SatelliteZenithAngle",
"SatelliteAzimuthAngle"]
if is_dnb:
angles += ["LunarZenithAngle", "LunarAzimuthAngle"]
for k in angles:
k = data_var_prefix + "/" + k
file_content[k] = lon_data # close enough to SZA
file_content[k] = np.repeat([file_content[k]], total_rows, axis=0)
file_content[k + "/shape"] = new_shape
[docs]
@staticmethod
def _add_geo_ref(file_content, filename):
if filename[:3] == "SVI":
geo_prefix = "GIMGO"
elif filename[:3] == "SVM":
geo_prefix = "GMODO"
else:
geo_prefix = None
if geo_prefix:
file_content["/attr/N_GEO_Ref"] = geo_prefix + filename[5:]
[docs]
@staticmethod
def _convert_numpy_content_to_dataarray(final_content):
import dask.array as da
from xarray import DataArray
for key, val in final_content.items():
if isinstance(val, np.ndarray):
val = da.from_array(val, chunks=val.shape)
if val.ndim > 1:
final_content[key] = DataArray(val, dims=("y", "x"))
else:
final_content[key] = DataArray(val)
[docs]
def get_test_content(self, filename, filename_info, filetype_info):
"""Mimic reader input file content."""
final_content = {}
for dataset in self.datasets:
dataset_group = DATASET_KEYS[dataset]
prefix1 = "Data_Products/{dataset_group}".format(dataset_group=dataset_group)
prefix2 = "{prefix}/{dataset_group}_Aggr".format(prefix=prefix1, dataset_group=dataset_group)
prefix3 = "All_Data/{dataset_group}_All".format(dataset_group=dataset_group)
file_content = {}
self._add_basic_metadata_to_file_content(file_content, filename_info, self._num_test_granules)
self._add_granule_specific_info_to_file_content(file_content, dataset_group,
self._num_test_granules, self._num_scans_per_gran,
prefix1)
self._add_geo_ref(file_content, filename)
for k, v in list(file_content.items()):
file_content[k.format(prefix1=prefix1, prefix2=prefix2, prefix3=prefix3)] = v
if filename[:3] in ["SVM", "SVI", "SVD"]:
self._add_data_info_to_file_content(file_content, filename, prefix3,
self._num_test_granules)
elif filename[0] == "G":
self._add_geolocation_info_to_file_content(file_content, filename, prefix3,
self._num_test_granules)
final_content.update(file_content)
self._convert_numpy_content_to_dataarray(final_content)
return final_content
[docs]
@contextmanager
def touch_geo_files(*prefixes):
"""Create and then remove VIIRS SDR geolocation files."""
geofiles = [_touch_geo_file(prefix) for prefix in prefixes]
try:
yield geofiles
finally:
for filename in geofiles:
os.remove(filename)
[docs]
def _touch_geo_file(prefix):
geo_fn = prefix + "_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5"
open(geo_fn, "w")
return geo_fn
[docs]
class TestVIIRSSDRReader(unittest.TestCase):
"""Test VIIRS SDR Reader."""
yaml_file = "viirs_sdr.yaml"
[docs]
def _assert_reflectance_properties(self, data_arr, num_scans=16, with_area=True):
assert np.issubdtype(data_arr.dtype, np.float32)
assert data_arr.attrs["calibration"] == "reflectance"
assert data_arr.attrs["units"] == "%"
assert data_arr.attrs["rows_per_scan"] == num_scans
if with_area:
assert "area" in data_arr.attrs
assert data_arr.attrs["area"] is not None
assert data_arr.attrs["area"].shape == data_arr.shape
else:
assert "area" not in data_arr.attrs
[docs]
def _assert_bt_properties(self, data_arr, num_scans=16, with_area=True):
assert np.issubdtype(data_arr.dtype, np.float32)
assert data_arr.attrs["calibration"] == "brightness_temperature"
assert data_arr.attrs["units"] == "K"
assert data_arr.attrs["rows_per_scan"] == num_scans
if with_area:
assert "area" in data_arr.attrs
assert data_arr.attrs["area"] is not None
assert data_arr.attrs["area"].shape == data_arr.shape
else:
assert "area" not in data_arr.attrs
[docs]
def _assert_dnb_radiance_properties(self, data_arr, with_area=True):
assert np.issubdtype(data_arr.dtype, np.float32)
assert data_arr.attrs["calibration"] == "radiance"
assert data_arr.attrs["units"] == "W m-2 sr-1"
assert data_arr.attrs["rows_per_scan"] == 16
if with_area:
assert "area" in data_arr.attrs
assert data_arr.attrs["area"] is not None
assert data_arr.attrs["area"].shape == data_arr.shape
else:
assert "area" not in data_arr.attrs
[docs]
def setUp(self):
"""Wrap HDF5 file handler with our own fake handler."""
from satpy._config import config_search_paths
from satpy.readers.viirs_atms_sdr_base import JPSS_SDR_FileHandler
self.reader_configs = config_search_paths(os.path.join("readers", self.yaml_file))
# http://stackoverflow.com/questions/12219967/how-to-mock-a-base-class-with-python-mock-library
self.p = mock.patch.object(JPSS_SDR_FileHandler, "__bases__", (FakeHDF5FileHandler2,))
self.fake_handler = self.p.start()
self.p.is_local = True
[docs]
def test_init(self):
"""Test basic init with no extra parameters."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
assert len(loadables) == 1
r.create_filehandlers(loadables)
# make sure we have some files
assert r.file_handlers
[docs]
def test_init_start_time_is_nodate(self):
"""Test basic init with start_time being set to the no-date 1/1-1958."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
with pytest.raises(ValueError, match="Datetime invalid 1958-01-01 00:00:00"):
_ = r.create_filehandlers([
"SVI01_npp_d19580101_t0000000_e0001261_b01708_c20120226002130255476_noaa_ops.h5",
])
[docs]
def test_init_start_time_beyond(self):
"""Test basic init with start_time after the provided files."""
from datetime import datetime
from satpy.readers import load_reader
r = load_reader(self.reader_configs,
filter_parameters={
"start_time": datetime(2012, 2, 26)
})
fhs = r.create_filehandlers([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
assert len(fhs) == 0
[docs]
def test_init_end_time_beyond(self):
"""Test basic init with end_time before the provided files."""
from datetime import datetime
from satpy.readers import load_reader
r = load_reader(self.reader_configs,
filter_parameters={
"end_time": datetime(2012, 2, 24)
})
fhs = r.create_filehandlers([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
assert len(fhs) == 0
[docs]
def test_init_start_end_time(self):
"""Test basic init with end_time before the provided files."""
from datetime import datetime
from satpy.readers import load_reader
r = load_reader(self.reader_configs,
filter_parameters={
"start_time": datetime(2012, 2, 24),
"end_time": datetime(2012, 2, 26)
})
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
assert len(loadables) == 1
r.create_filehandlers(loadables)
# make sure we have some files
assert r.file_handlers
[docs]
def test_load_all_m_reflectances_no_geo(self):
"""Load all M band reflectances with no geo files provided."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["M01",
"M02",
"M03",
"M04",
"M05",
"M06",
"M07",
"M08",
"M09",
"M10",
"M11",
])
assert len(ds) == 11
for d in ds.values():
self._assert_reflectance_properties(d, with_area=False)
[docs]
def test_load_all_m_reflectances_find_geo(self):
"""Load all M band reflectances with geo files not specified but existing."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
with touch_geo_files("GMTCO", "GMODO") as (geo_fn1, geo_fn2):
r.create_filehandlers(loadables)
ds = r.load(["M01",
"M02",
"M03",
"M04",
"M05",
"M06",
"M07",
"M08",
"M09",
"M10",
"M11",
])
assert len(ds) == 11
for d in ds.values():
self._assert_reflectance_properties(d, with_area=True)
[docs]
def test_load_all_m_reflectances_provided_geo(self):
"""Load all M band reflectances with geo files provided."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMTCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
with touch_geo_files("GMTCO", "GMODO") as (geo_fn1, geo_fn2):
r.create_filehandlers(loadables)
ds = r.load(["M01",
"M02",
"M03",
"M04",
"M05",
"M06",
"M07",
"M08",
"M09",
"M10",
"M11",
])
assert len(ds) == 11
for d in ds.values():
self._assert_reflectance_properties(d, with_area=True)
assert d.attrs["area"].lons.min() == 5
assert d.attrs["area"].lats.min() == 45
assert d.attrs["area"].lons.attrs["rows_per_scan"] == 16
assert d.attrs["area"].lats.attrs["rows_per_scan"] == 16
[docs]
def test_load_all_m_reflectances_use_nontc(self):
"""Load all M band reflectances but use non-TC geolocation."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs, use_tc=False)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMTCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMODO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
with touch_geo_files("GMTCO", "GMODO") as (geo_fn1, geo_fn2):
r.create_filehandlers(loadables, {"use_tc": False})
ds = r.load(["M01",
"M02",
"M03",
"M04",
"M05",
"M06",
"M07",
"M08",
"M09",
"M10",
"M11",
])
assert len(ds) == 11
for d in ds.values():
self._assert_reflectance_properties(d, with_area=True)
assert d.attrs["area"].lons.min() == 15
assert d.attrs["area"].lats.min() == 55
assert d.attrs["area"].lons.attrs["rows_per_scan"] == 16
assert d.attrs["area"].lats.attrs["rows_per_scan"] == 16
[docs]
def test_load_all_m_reflectances_use_nontc2(self):
"""Load all M band reflectances but use non-TC geolocation because TC isn't available."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs, use_tc=None)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMODO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
with touch_geo_files("GMODO") as (geo_fn2,):
r.create_filehandlers(loadables, {"use_tc": None})
ds = r.load(["M01",
"M02",
"M03",
"M04",
"M05",
"M06",
"M07",
"M08",
"M09",
"M10",
"M11",
])
assert len(ds) == 11
for d in ds.values():
self._assert_reflectance_properties(d, with_area=True)
assert d.attrs["area"].lons.min() == 15
assert d.attrs["area"].lats.min() == 55
assert d.attrs["area"].lons.attrs["rows_per_scan"] == 16
assert d.attrs["area"].lats.attrs["rows_per_scan"] == 16
[docs]
def test_load_all_m_bts(self):
"""Load all M band brightness temperatures."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVM12_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM13_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM14_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM15_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM16_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMTCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["M12",
"M13",
"M14",
"M15",
"M16",
])
assert len(ds) == 5
for d in ds.values():
self._assert_bt_properties(d, with_area=True)
[docs]
def test_load_dnb_sza_no_factors(self):
"""Load DNB solar zenith angle with no scaling factors.
The angles in VIIRS SDRs should never have scaling factors so we test
it that way.
"""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"GDNBO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables, {"include_factors": False})
ds = r.load(["dnb_solar_zenith_angle",
"dnb_solar_azimuth_angle",
"dnb_satellite_zenith_angle",
"dnb_satellite_azimuth_angle",
"dnb_lunar_zenith_angle",
"dnb_lunar_azimuth_angle"])
assert len(ds) == 6
for d in ds.values():
assert np.issubdtype(d.dtype, np.float32)
assert d.attrs["units"] == "degrees"
assert d.attrs["rows_per_scan"] == 16
assert "area" in d.attrs
assert d.attrs["area"] is not None
[docs]
def test_load_all_m_radiances(self):
"""Load all M band radiances."""
from satpy.readers import load_reader
from satpy.tests.utils import make_dsq
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVM01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM06_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM07_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM08_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM09_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM10_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM11_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM12_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM13_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM14_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM15_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVM16_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GMTCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load([
make_dsq(name="M01", calibration="radiance"),
make_dsq(name="M02", calibration="radiance"),
make_dsq(name="M03", calibration="radiance"),
make_dsq(name="M04", calibration="radiance"),
make_dsq(name="M05", calibration="radiance"),
make_dsq(name="M06", calibration="radiance"),
make_dsq(name="M07", calibration="radiance"),
make_dsq(name="M08", calibration="radiance"),
make_dsq(name="M09", calibration="radiance"),
make_dsq(name="M10", calibration="radiance"),
make_dsq(name="M11", calibration="radiance"),
make_dsq(name="M12", calibration="radiance"),
make_dsq(name="M13", calibration="radiance"),
make_dsq(name="M14", calibration="radiance"),
make_dsq(name="M15", calibration="radiance"),
make_dsq(name="M16", calibration="radiance"),
])
assert len(ds) == 16
for d in ds.values():
assert np.issubdtype(d.dtype, np.float32)
assert d.attrs["calibration"] == "radiance"
assert d.attrs["units"] == "W m-2 um-1 sr-1"
assert d.attrs["rows_per_scan"] == 16
assert "area" in d.attrs
assert d.attrs["area"] is not None
[docs]
def test_load_dnb(self):
"""Load DNB dataset."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVDNB_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GDNBO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["DNB"])
assert len(ds) == 1
for d in ds.values():
data = d.values
# default scale factors are 2 and offset 1
# multiply DNB by 10000 should mean the first value of 0 should be:
# data * factor * 10000 + offset * 10000
# 0 * 2 * 10000 + 1 * 10000 => 10000
assert data[0, 0] == 10000
# the second value of 1 should be:
# 1 * 2 * 10000 + 1 * 10000 => 30000
assert data[0, 1] == 30000
self._assert_dnb_radiance_properties(d, with_area=True)
[docs]
def test_load_dnb_no_factors(self):
"""Load DNB dataset with no provided scale factors."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVDNB_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GDNBO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables, {"include_factors": False})
ds = r.load(["DNB"])
assert len(ds) == 1
for d in ds.values():
data = d.values
# no scale factors, default factor 1 and offset 0
# multiply DNB by 10000 should mean the first value of 0 should be:
# data * factor * 10000 + offset * 10000
# 0 * 1 * 10000 + 0 * 10000 => 0
assert data[0, 0] == 0
# the second value of 1 should be:
# 1 * 1 * 10000 + 0 * 10000 => 10000
assert data[0, 1] == 10000
self._assert_dnb_radiance_properties(d, with_area=True)
[docs]
def test_load_i_no_files(self):
"""Load I01 when only DNB files are provided."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVDNB_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GDNBO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
assert "I01" not in [x["name"] for x in r.available_dataset_ids]
ds = r.load(["I01"])
assert len(ds) == 0
[docs]
def test_load_all_i_reflectances_provided_geo(self):
"""Load all I band reflectances with geo files provided."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GITCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["I01",
"I02",
"I03",
])
assert len(ds) == 3
for d in ds.values():
self._assert_reflectance_properties(d, num_scans=32)
assert d.attrs["area"].lons.min() == 5
assert d.attrs["area"].lats.min() == 45
assert d.attrs["area"].lons.attrs["rows_per_scan"] == 32
assert d.attrs["area"].lats.attrs["rows_per_scan"] == 32
[docs]
def test_load_all_i_bts(self):
"""Load all I band brightness temperatures."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GITCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["I04",
"I05",
])
assert len(ds) == 2
for d in ds.values():
self._assert_bt_properties(d, num_scans=32)
[docs]
def test_load_all_i_radiances(self):
"""Load all I band radiances."""
from satpy.readers import load_reader
from satpy.tests.utils import make_dsq
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI02_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI03_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI04_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"SVI05_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
"GITCO_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load([
make_dsq(name="I01", calibration="radiance"),
make_dsq(name="I02", calibration="radiance"),
make_dsq(name="I03", calibration="radiance"),
make_dsq(name="I04", calibration="radiance"),
make_dsq(name="I05", calibration="radiance"),
])
assert len(ds) == 5
for d in ds.values():
assert np.issubdtype(d.dtype, np.float32) is True
assert d.attrs["calibration"] == "radiance"
assert d.attrs["units"] == "W m-2 um-1 sr-1"
assert d.attrs["rows_per_scan"] == 32
assert "area" in d.attrs
assert d.attrs["area"] is not None
[docs]
class FakeHDF5FileHandlerAggr(FakeHDF5FileHandler2):
"""Swap-in HDF5 File Handler with 4 VIIRS Granules per file."""
_num_test_granules = 4
_num_scans_per_gran = [48] * 4
[docs]
class TestAggrVIIRSSDRReader(unittest.TestCase):
"""Test VIIRS SDR Reader."""
yaml_file = "viirs_sdr.yaml"
[docs]
def setUp(self):
"""Wrap HDF5 file handler with our own fake handler."""
from satpy._config import config_search_paths
from satpy.readers.viirs_sdr import VIIRSSDRFileHandler
self.reader_configs = config_search_paths(os.path.join("readers", self.yaml_file))
# http://stackoverflow.com/questions/12219967/how-to-mock-a-base-class-with-python-mock-library
self.p = mock.patch.object(VIIRSSDRFileHandler, "__bases__", (FakeHDF5FileHandlerAggr,))
self.fake_handler = self.p.start()
self.p.is_local = True
[docs]
def test_bounding_box(self):
"""Test bounding box."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
# make sure we have some files
expected_lons = [
72.50243, 64.17125, 59.15234, 59.386833, 55.770416, 53.38952, 53.52594, 51.685738, 50.439102, 50.51393,
49.566296, 48.865967, 18.96082, -4.0238385, -7.05221, -10.247547, -13.951393, -18.062435, -22.608992,
-27.620953, -33.091743, -39.28113, -17.749891
]
expected_lats = [
81.67615, 79.49934, 77.278656, 77.393425, 74.977875, 72.62976, 72.74879, 70.2493, 67.84738, 67.969505,
65.545685, 63.103046, 61.853905, 55.169273, 57.062447, 58.77254, 60.465942, 62.036346, 63.465122,
64.72178, 65.78417, 66.66166, 79.00025
]
lons, lats = r.file_handlers["generic_file"][0].get_bounding_box()
np.testing.assert_allclose(lons, expected_lons)
np.testing.assert_allclose(lats, expected_lats)
[docs]
class FakeShortHDF5FileHandlerAggr(FakeHDF5FileHandler2):
"""Fake file that has less scans than usual in a couple granules."""
_num_test_granules = 3
_num_scans_per_gran = [47, 48, 47]
[docs]
class TestShortAggrVIIRSSDRReader(unittest.TestCase):
"""Test VIIRS SDR Reader with a file that has truncated granules."""
yaml_file = "viirs_sdr.yaml"
[docs]
def setUp(self):
"""Wrap HDF5 file handler with our own fake handler."""
from satpy._config import config_search_paths
from satpy.readers.viirs_atms_sdr_base import JPSS_SDR_FileHandler
self.reader_configs = config_search_paths(os.path.join("readers", self.yaml_file))
# http://stackoverflow.com/questions/12219967/how-to-mock-a-base-class-with-python-mock-library
self.p = mock.patch.object(JPSS_SDR_FileHandler, "__bases__", (FakeShortHDF5FileHandlerAggr,))
self.fake_handler = self.p.start()
self.p.is_local = True
[docs]
def test_load_truncated_band(self):
"""Test loading a single truncated band."""
from satpy.readers import load_reader
r = load_reader(self.reader_configs)
loadables = r.select_files_from_pathnames([
"SVI01_npp_d20120225_t1801245_e1802487_b01708_c20120226002130255476_noaa_ops.h5",
])
r.create_filehandlers(loadables)
ds = r.load(["I01"])
assert len(ds) == 1
i01_data = ds["I01"].compute()
expected_rows = sum(FakeShortHDF5FileHandlerAggr._num_scans_per_gran) * DEFAULT_FILE_SHAPE[0]
assert i01_data.shape == (expected_rows, 300)