#!/usr/bin/env python
# encoding: utf-8
# Copyright (C) 2022 Space Science and Engineering Center (SSEC),
# University of Wisconsin-Madison.
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
#
# This file is part of the polar2grid software package. Polar2grid takes
# satellite observation data, remaps it, and writes it to a file format for
# input into another program.
# Documentation: http://www.ssec.wisc.edu/software/polar2grid/
"""Test fixtures representing AVHRR swath data."""
from __future__ import annotations
import dask.array as da
import numpy as np
import pytest
import xarray as xr
from pyresample import SwathDefinition
from satpy import Scene
from satpy.tests.utils import make_dataid
from ._fixture_utils import START_TIME, _TestingScene, generate_lonlat_data
AVHRR_SHAPE = (2361, 2048)
AVHRR_CHUNKS = AVHRR_SHAPE
AVHRR_IDS = [
make_dataid(name="1", wavelength=(0.58, 0.63, 0.68), resolution=1050, calibration="reflectance"),
]
[docs]
@pytest.fixture(scope="session")
def avhrr_l1b_swath_def() -> SwathDefinition:
lons, lats = generate_lonlat_data(AVHRR_SHAPE)
lons_data_arr = xr.DataArray(
da.from_array(lons, chunks=AVHRR_CHUNKS),
dims=("y", "x"),
attrs={
"platform_name": "noaa19",
"sensor": "avhrr-3",
"start_time": START_TIME,
"end_time": START_TIME,
"resolution": 1050,
"reader": "avhrr_l1b_aapp",
},
)
lats_data_arr = xr.DataArray(
da.from_array(lats, chunks=AVHRR_CHUNKS),
dims=("y", "x"),
attrs={
"platform_name": "noaa19",
"sensor": "avhrr-3",
"start_time": START_TIME,
"end_time": START_TIME,
"resolution": 1050,
"reader": "avhrr_l1b_aapp",
},
)
return SwathDefinition(lons_data_arr, lats_data_arr)
[docs]
@pytest.fixture
def avhrr_l1b_1_data_array(avhrr_l1b_swath_def) -> xr.DataArray:
return xr.DataArray(
da.zeros(AVHRR_SHAPE, dtype=np.float32),
dims=("y", "x"),
attrs={
"area": avhrr_l1b_swath_def,
"platform_name": "noaa19",
"sensor": "avhrr-3",
"name": "1",
"start_time": START_TIME,
"end_time": START_TIME,
"resolution": 1050,
"reader": "avhrr_l1b_aapp",
"calibration": "reflectance",
"standard_name": "toa_bidirectional_reflectance",
"units": "%",
},
)
[docs]
@pytest.fixture
def avhrr_l1b_1_scene(avhrr_l1b_1_data_array) -> Scene:
scn = _TestingScene(
reader="avhrr_l1b_aapp",
filenames=["/fake/filename"],
data_array_dict={
AVHRR_IDS[0]: avhrr_l1b_1_data_array.copy(),
},
all_dataset_ids=AVHRR_IDS,
available_dataset_ids=AVHRR_IDS[:1],
)
return scn
