Source code for polar2grid.tests.test_writers.test_base

#!/usr/bin/env python3
# encoding: utf-8
# Copyright (C) 2021 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/
"""Tests for base/shared logic."""

import importlib
import os

import pytest

PKG_ROOT = os.path.realpath(os.path.join(os.path.dirname(__file__), "..", ".."))
WRITER_BASE = os.path.join(PKG_ROOT, "writers")
ALL_WRITERS = [os.path.splitext(writer_fn)[0] for writer_fn in os.listdir(WRITER_BASE) if not writer_fn.startswith("_")]


@pytest.mark.parametrize("writer_name", ALL_WRITERS)
def test_writer_imports(writer_name):
    mod = importlib.import_module(f"polar2grid.writers.{writer_name}")
    assert hasattr(mod, "DEFAULT_OUTPUT_FILENAMES")
    assert isinstance(mod.DEFAULT_OUTPUT_FILENAMES, dict)
    assert "polar2grid" in mod.DEFAULT_OUTPUT_FILENAMES
    assert "geo2grid" in mod.DEFAULT_OUTPUT_FILENAMES


@pytest.mark.parametrize("writer_name", ALL_WRITERS)
def test_reader_parser_arguments(writer_name):
    import argparse

    mod = importlib.import_module(f"polar2grid.writers.{writer_name}")
    assert hasattr(mod, "add_writer_argument_groups")

    parser = argparse.ArgumentParser()
    groups = mod.add_writer_argument_groups(parser)
    assert len(groups) == 2