Source code for satpy.tests.reader_tests.test_nwcsaf_nc

#!/usr/bin/env python
# Copyright (c) 2018, 2020 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/>.
"""Unittests for NWC SAF reader."""

import unittest
from unittest import mock

PROJ_KM = {'gdal_projection': '+proj=geos +a=6378.137000 +b=6356.752300 +lon_0=0.000000 +h=35785.863000',
           'gdal_xgeo_up_left': -5569500.0,
           'gdal_ygeo_up_left': 5437500.0,
           'gdal_xgeo_low_right': 5566500.0,
           'gdal_ygeo_low_right': 2653500.0}
PROJ = {'gdal_projection': '+proj=geos +a=6378137.000 +b=6356752.300 +lon_0=0.000000 +h=35785863.000',
        'gdal_xgeo_up_left': -5569500.0,
        'gdal_ygeo_up_left': 5437500.0,
        'gdal_xgeo_low_right': 5566500.0,
        'gdal_ygeo_low_right': 2653500.0}


[docs]class TestNcNWCSAF(unittest.TestCase): """Test the NcNWCSAF reader."""
[docs] @mock.patch('satpy.readers.nwcsaf_nc.unzip_file') @mock.patch('satpy.readers.nwcsaf_nc.xr') def setUp(self, xr_, unzip): """Set up the test case.""" from satpy.readers.nwcsaf_nc import NcNWCSAF xr_.return_value = mock.Mock(attrs={}) unzip.return_value = '' self.scn = NcNWCSAF('filename', {}, {})
[docs] def test_sensor_name(self): """Test that the correct sensor name is being set.""" self.scn.set_platform_and_sensor(platform_name='Metop-B') self.assertEqual(self.scn.sensor, set(['avhrr-3'])) self.assertEqual(self.scn.sensor_names, set(['avhrr-3'])) self.scn.set_platform_and_sensor(platform_name='NOAA-20') self.assertEqual(self.scn.sensor, set(['viirs'])) self.assertEqual(self.scn.sensor_names, set(['viirs'])) self.scn.set_platform_and_sensor(platform_name='Himawari-8') self.assertEqual(self.scn.sensor, set(['ahi'])) self.assertEqual(self.scn.sensor_names, set(['ahi'])) self.scn.set_platform_and_sensor(sat_id='GOES16') self.assertEqual(self.scn.sensor, set(['abi'])) self.assertEqual(self.scn.sensor_names, set(['abi'])) self.scn.set_platform_and_sensor(platform_name='GOES-17') self.assertEqual(self.scn.sensor, set(['abi'])) self.assertEqual(self.scn.sensor_names, set(['abi'])) self.scn.set_platform_and_sensor(sat_id='MSG4') self.assertEqual(self.scn.sensor, set(['seviri'])) self.scn.set_platform_and_sensor(platform_name='Meteosat-11') self.assertEqual(self.scn.sensor, set(['seviri'])) self.assertEqual(self.scn.sensor_names, set(['seviri']))
[docs] def test_get_area_def(self): """Test that get_area_def() returns proper area.""" dsid = {'name': 'foo'} self.scn.nc[dsid['name']].shape = (5, 10) # a, b and h in kilometers self.scn.nc.attrs = PROJ_KM _check_area_def(self.scn.get_area_def(dsid)) # a, b and h in meters self.scn.nc.attrs = PROJ _check_area_def(self.scn.get_area_def(dsid))
[docs] def test_scale_dataset_attr_removal(self): """Test the scaling of the dataset and removal of obsolete attributes.""" import numpy as np import xarray as xr attrs = {'scale_factor': np.array(10), 'add_offset': np.array(20)} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [30, 40, 50]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs)
[docs] def test_scale_dataset_floating(self): """Test the scaling of the dataset with floating point values.""" import numpy as np import xarray as xr attrs = {'scale_factor': np.array(1.5), 'add_offset': np.array(2.5), '_FillValue': 1} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [np.nan, 5.5, 7]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs) attrs = {'scale_factor': np.array(1.5), 'add_offset': np.array(2.5), 'valid_min': 1.1} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [np.nan, 5.5, 7]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs) attrs = {'scale_factor': np.array(1.5), 'add_offset': np.array(2.5), 'valid_max': 2.1} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [4, 5.5, np.nan]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs) attrs = {'scale_factor': np.array(1.5), 'add_offset': np.array(2.5), 'valid_range': (1.1, 2.1)} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [np.nan, 5.5, np.nan]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs) # CTTH NWCSAF/Geo v2016/v2018: attrs = {'scale_factor': np.array(1.), 'add_offset': np.array(-2000.), 'valid_range': (0., 27000.)} var = xr.DataArray([1, 2, 3], attrs=attrs) var = self.scn.scale_dataset('dummy', var, 'dummy') np.testing.assert_allclose(var, [-1999., -1998., -1997.]) self.assertNotIn('scale_factor', var.attrs) self.assertNotIn('add_offset', var.attrs) self.assertEqual(var.attrs['valid_range'][0], -2000.) self.assertEqual(var.attrs['valid_range'][1], 25000.)
def _check_area_def(area_definition): correct_h = float(PROJ['gdal_projection'].split('+h=')[-1]) correct_a = float(PROJ['gdal_projection'].split('+a=')[-1].split()[0]) assert area_definition.proj_dict['h'] == correct_h assert area_definition.proj_dict['a'] == correct_a assert area_definition.proj_dict['units'] == 'm' correct_extent = (PROJ["gdal_xgeo_up_left"], PROJ["gdal_ygeo_low_right"], PROJ["gdal_xgeo_low_right"], PROJ["gdal_ygeo_up_left"]) assert area_definition.area_extent == correct_extent