radar_fun.py

import netCDF4
import numpy as np


class RadarFun:
    """Common tests for all radars."""

    def __init__(self, nc: netCDF4.Dataset, site_meta: dict, date: str, uuid):
        self.nc = nc
        self.site_meta = site_meta
        self.date = date
        self.uuid = uuid

    def test_variable_names(self):
        keys = {
            "Zh",
            "v",
            "radar_frequency",
            "range",
            "zenith_angle",
            "height",
        }
        for key in keys:
            assert key in self.nc.variables

    def test_axis(self):
        assert self.nc.variables["range"].axis == "Z"
        for key in self.nc.variables.keys():
            if key not in ("time", "range"):
                assert hasattr(self.nc.variables[key], "axis") is False

    def test_variable_values(self):
        assert 0 <= np.all(self.nc.variables["zenith_angle"][:]) < 10
        assert np.all(
            (
                self.nc.variables["height"][:]
                - self.site_meta["altitude"]
                - self.nc.variables["range"][:]
            )
            <= 1e-3,
        )

    def test_global_attributes(self):
        assert self.nc.cloudnet_file_type == "radar"