Source code for satpy.tests.reader_tests.test_goes_imager_hrit

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2018 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/>.
"""The hrit msg reader tests package."""

import datetime
import unittest
from unittest import mock

import numpy as np
from pyresample.utils import proj4_radius_parameters
from xarray import DataArray

from satpy.readers.goes_imager_hrit import (
    ALTITUDE,
    EQUATOR_RADIUS,
    POLE_RADIUS,
    HRITGOESFileHandler,
    HRITGOESPrologueFileHandler,
    make_gvar_float,
    make_sgs_time,
    sgs_time,
)
from satpy.tests.utils import make_dataid


[docs] class TestGVARFloat(unittest.TestCase): """GVAR float tester."""
[docs] def test_fun(self): """Test function.""" test_data = [(-1.0, b"\xbe\xf0\x00\x00"), (-0.1640625, b"\xbf\xd6\x00\x00"), (0.0, b"\x00\x00\x00\x00"), (0.1640625, b"\x40\x2a\x00\x00"), (1.0, b"\x41\x10\x00\x00"), (100.1640625, b"\x42\x64\x2a\x00")] for expected, str_val in test_data: val = np.frombuffer(str_val, dtype='>i4') self.assertEqual(expected, make_gvar_float(val))
[docs] class TestMakeSGSTime(unittest.TestCase): """SGS Time tester."""
[docs] def test_fun(self): """Encode the test time.""" # 2018-129 (may 9th), 21:33:27.999 tcds = np.array([(32, 24, 18, 146, 19, 50, 121, 153)], dtype=sgs_time) expected = datetime.datetime(2018, 5, 9, 21, 33, 27, 999000) self.assertEqual(make_sgs_time(tcds[0]), expected)
test_pro = {'TISTR': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TCurr': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TCLMT': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'SubSatLongitude': 100.1640625, 'TCHED': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TLTRL': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TIPFS': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TISPC': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'ReferenceLatitude': 0.0, 'TIIRT': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TLHED': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TIVIT': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'SubSatLatitude': 0.0, 'TIECL': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'ReferenceLongitude': 100.1640625, 'TCTRL': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TLRAN': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TINFS': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TIBBC': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'TIONA': datetime.datetime(2018, 5, 9, 21, 33, 27, 999000), 'ReferenceDistance': 100.1640625, 'SatelliteID': 15}
[docs] class TestHRITGOESPrologueFileHandler(unittest.TestCase): """Test the HRITFileHandler."""
[docs] @mock.patch('satpy.readers.goes_imager_hrit.recarray2dict') @mock.patch('satpy.readers.goes_imager_hrit.np.fromfile') @mock.patch('satpy.readers.goes_imager_hrit.HRITFileHandler.__init__') def test_init(self, new_fh_init, fromfile, recarray2dict): """Setup the hrit file handler for testing.""" recarray2dict.side_effect = lambda x: x[0] new_fh_init.return_value.filename = 'filename' HRITGOESPrologueFileHandler.filename = 'filename' HRITGOESPrologueFileHandler.mda = {'total_header_length': 1} ret = {} the_time = np.array([(32, 24, 18, 146, 19, 50, 121, 153)], dtype=sgs_time)[0] for key in ['TCurr', 'TCHED', 'TCTRL', 'TLHED', 'TLTRL', 'TIPFS', 'TINFS', 'TISPC', 'TIECL', 'TIBBC', 'TISTR', 'TLRAN', 'TIIRT', 'TIVIT', 'TCLMT', 'TIONA']: ret[key] = the_time ret['SubSatLatitude'] = np.frombuffer(b"\x00\x00\x00\x00", dtype='>i4')[0] ret['ReferenceLatitude'] = np.frombuffer(b"\x00\x00\x00\x00", dtype='>i4')[0] ret['SubSatLongitude'] = np.frombuffer(b"\x42\x64\x2a\x00", dtype='>i4')[0] ret['ReferenceLongitude'] = np.frombuffer(b"\x42\x64\x2a\x00", dtype='>i4')[0] ret['ReferenceDistance'] = np.frombuffer(b"\x42\x64\x2a\x00", dtype='>i4')[0] ret['SatelliteID'] = 15 fromfile.return_value = [ret] m = mock.mock_open() with mock.patch('satpy.readers.goes_imager_hrit.open', m, create=True) as newopen: newopen.return_value.__enter__.return_value.seek.return_value = 1 self.reader = HRITGOESPrologueFileHandler( 'filename', {'platform_shortname': 'GOES15', 'start_time': datetime.datetime(2016, 3, 3, 0, 0), 'service': 'test_service'}, {'filetype': 'info'}) self.assertEqual(test_pro, self.reader.prologue)
[docs] class TestHRITGOESFileHandler(unittest.TestCase): """Test the HRITFileHandler."""
[docs] @mock.patch('satpy.readers.goes_imager_hrit.HRITFileHandler.__init__') def setUp(self, new_fh_init): """Set up the hrit file handler for testing.""" blob = '$HALFTONE:=10\r\n_NAME:=albedo\r\n_UNIT:=percent\r\n0:=0.0\r\n1023:=100.0\r\n'.encode() mda = {'projection_parameters': {'SSP_longitude': -123.0}, 'spectral_channel_id': 1, 'image_data_function': blob} HRITGOESFileHandler.filename = 'filename' HRITGOESFileHandler.mda = mda self.prologue = mock.MagicMock() self.prologue.prologue = test_pro self.reader = HRITGOESFileHandler('filename', {}, {}, self.prologue)
[docs] def test_init(self): """Test the init.""" blob = '$HALFTONE:=10\r\n_NAME:=albedo\r\n_UNIT:=percent\r\n0:=0.0\r\n1023:=100.0\r\n'.encode() mda = {'spectral_channel_id': 1, 'projection_parameters': {'SSP_longitude': 100.1640625}, 'image_data_function': blob} self.assertEqual(self.reader.mda, mda)
[docs] @mock.patch('satpy.readers.goes_imager_hrit.HRITFileHandler.get_dataset') def test_get_dataset(self, base_get_dataset): """Test get_dataset.""" key = make_dataid(name="CH1", calibration='reflectance') base_get_dataset.return_value = DataArray(np.arange(25).reshape(5, 5)) res = self.reader.get_dataset(key, {}) expected = np.array([[np.nan, 0.097752, 0.195503, 0.293255, 0.391007], [0.488759, 0.58651, 0.684262, 0.782014, 0.879765], [0.977517, 1.075269, 1.173021, 1.270772, 1.368524], [1.466276, 1.564027, 1.661779, 1.759531, 1.857283], [1.955034, 2.052786, 2.150538, 2.248289, 2.346041]]) self.assertTrue(np.allclose(res.values, expected, equal_nan=True)) self.assertEqual(res.attrs['units'], '%') self.assertDictEqual(res.attrs['orbital_parameters'], {'projection_longitude': self.reader.mda['projection_parameters']['SSP_longitude'], 'projection_latitude': 0.0, 'projection_altitude': ALTITUDE})
[docs] def test_get_area_def(self): """Test getting the area definition.""" self.reader.mda.update({ 'cfac': 10216334, 'lfac': 10216334, 'coff': 1408.0, 'loff': 944.0, 'number_of_lines': 464, 'number_of_columns': 2816 }) dsid = make_dataid(name="CH1", calibration='reflectance', resolution=3000) area = self.reader.get_area_def(dsid) a, b = proj4_radius_parameters(area.proj_dict) assert a == EQUATOR_RADIUS assert b == POLE_RADIUS assert area.proj_dict['h'] == ALTITUDE assert area.proj_dict['lon_0'] == 100.1640625 assert area.proj_dict['proj'] == 'geos' assert area.proj_dict['units'] == 'm' assert area.width == 2816 assert area.height == 464 assert area.area_id == 'goes-15_goes_imager_fd_3km' area_extent_exp = (-5639254.900260435, 1925159.4881528523, 5643261.475678028, 3784210.48191544) np.testing.assert_allclose(area.area_extent, area_extent_exp)