#!/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)