class TestNcNWCSAF(unittest.TestCase):
@mock.patch('satpy.readers.nwcsaf_nc.unzip_file')
@mock.patch('satpy.readers.nwcsaf_nc.xr')
def setUp(self, xr_, unzip):
from satpy.readers.nwcsaf_nc import NcNWCSAF
xr_.return_value = mock.Mock(attrs={})
unzip.return_value = ''
self.scn = NcNWCSAF('filename', {}, {})
def tearDown(self):
pass
def test_get_projection(self):
# a, b and h in kilometers
self.scn.nc.attrs = PROJ_KM
proj_str, area_extent = self.scn._get_projection()
self.assertTrue('+units=km' in proj_str)
self.assertAlmostEqual(area_extent[0],
PROJ_KM['gdal_xgeo_up_left'] / 1000.)
self.assertAlmostEqual(area_extent[1],
PROJ_KM['gdal_ygeo_low_right'] / 1000.)
self.assertAlmostEqual(area_extent[2],
PROJ_KM['gdal_xgeo_low_right'] / 1000.)
self.assertAlmostEqual(area_extent[3],
PROJ_KM['gdal_ygeo_up_left'] / 1000.)
# a, b and h in meters
self.scn.nc.attrs = PROJ
proj_str, area_extent = self.scn._get_projection()
self.assertTrue('+units=m' in proj_str)
self.assertAlmostEqual(area_extent[0], PROJ['gdal_xgeo_up_left'])
self.assertAlmostEqual(area_extent[1], PROJ['gdal_ygeo_low_right'])
self.assertAlmostEqual(area_extent[2], PROJ['gdal_xgeo_low_right'])
self.assertAlmostEqual(area_extent[3], PROJ['gdal_ygeo_up_left'])
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', {}, {})
class TestNcNWCSAF(unittest.TestCase):
"""Test the NcNWCSAF reader."""
@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', {}, {})
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']))
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))
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)
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.)
class TestNcNWCSAF(unittest.TestCase):
"""Test the NcNWCSAF reader."""
@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', {}, {})
def test_get_projection(self):
"""Test generation of the navigation info."""
# a, b and h in kilometers
self.scn.nc.attrs = PROJ_KM
proj_str, area_extent = self.scn._get_projection()
self.assertTrue('+units=km' in proj_str)
self.assertAlmostEqual(area_extent[0],
PROJ_KM['gdal_xgeo_up_left'] / 1000.)
self.assertAlmostEqual(area_extent[1],
PROJ_KM['gdal_ygeo_low_right'] / 1000.)
self.assertAlmostEqual(area_extent[2],
PROJ_KM['gdal_xgeo_low_right'] / 1000.)
self.assertAlmostEqual(area_extent[3],
PROJ_KM['gdal_ygeo_up_left'] / 1000.)
# a, b and h in meters
self.scn.nc.attrs = PROJ
proj_str, area_extent = self.scn._get_projection()
self.assertTrue('+units=m' in proj_str)
self.assertAlmostEqual(area_extent[0], PROJ['gdal_xgeo_up_left'])
self.assertAlmostEqual(area_extent[1], PROJ['gdal_ygeo_low_right'])
self.assertAlmostEqual(area_extent[2], PROJ['gdal_xgeo_low_right'])
self.assertAlmostEqual(area_extent[3], PROJ['gdal_ygeo_up_left'])
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)
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.)