Source code for satpy.tests.compositor_tests.test_sar

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 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/>.
"""Tests for SAR compositors."""

import unittest


[docs] class TestSARComposites(unittest.TestCase): """Test SAR-specific composites."""
[docs] def test_sar_ice(self): """Test creating a the sar_ice composite.""" import dask.array as da import numpy as np import xarray as xr from satpy.composites.sar import SARIce rows = 2 cols = 2 comp = SARIce('sar_ice', prerequisites=('hh', 'hv'), standard_name='sar-ice') hh = xr.DataArray(da.zeros((rows, cols), chunks=25) + 2000, dims=('y', 'x'), attrs={'name': 'hh'}) hv = xr.DataArray(da.zeros((rows, cols), chunks=25) + 1000, dims=('y', 'x'), attrs={'name': 'hv'}) res = comp((hh, hv)) self.assertIsInstance(res, xr.DataArray) self.assertIsInstance(res.data, da.Array) self.assertEqual(res.attrs['name'], 'sar_ice') self.assertEqual(res.attrs['standard_name'], 'sar-ice') data = res.compute() np.testing.assert_allclose(data.sel(bands='R'), 31.58280822) np.testing.assert_allclose(data.sel(bands='G'), 159869.56789876) np.testing.assert_allclose(data.sel(bands='B'), 44.68138191)
[docs] def test_sar_ice_log(self): """Test creating a the sar_ice_log composite.""" import dask.array as da import numpy as np import xarray as xr from satpy.composites.sar import SARIceLog rows = 2 cols = 2 comp = SARIceLog('sar_ice_log', prerequisites=('hh', 'hv'), standard_name='sar-ice-log') hh = xr.DataArray(da.zeros((rows, cols), chunks=25) - 10, dims=('y', 'x'), attrs={'name': 'hh'}) hv = xr.DataArray(da.zeros((rows, cols), chunks=25) - 20, dims=('y', 'x'), attrs={'name': 'hv'}) res = comp((hh, hv)) self.assertIsInstance(res, xr.DataArray) self.assertIsInstance(res.data, da.Array) self.assertEqual(res.attrs['name'], 'sar_ice_log') self.assertEqual(res.attrs['standard_name'], 'sar-ice-log') data = res.compute() np.testing.assert_allclose(data.sel(bands='R'), -20) np.testing.assert_allclose(data.sel(bands='G'), -4.6) np.testing.assert_allclose(data.sel(bands='B'), -10)