#174 Update test for v2 specs module

Author: astropenguin

tests/test_data.py

@@ -0,0 +1,72 @@
+# standard library
+from dataclasses import dataclass
+from typing import Collection, Tuple, Union
+
+
+# dependencies
+import numpy as np
+from typing_extensions import Annotated as Ann, Literal as L
+from xarray_dataclasses.v2.typing import Attr, Coord, Coordof, Data
+
+
+# type hints
+X = L["x"]
+Y = L["y"]
+T = L["time"]
+_ = Tuple[()]
+
+
+# test data
+@dataclass
+class Longitude:
+    """Specification of longitude."""
+
+    data: Data[Tuple[X, Y], float]
+    units: Attr[str] = "deg"
+
+
+@dataclass
+class Latitude:
+    """Specification of latitude."""
+
+    data: Data[Tuple[X, Y], float]
+    units: Attr[str] = "deg"
+
+
+@dataclass
+class Weather:
+    """Weather information."""
+
+    temp: Ann[Data[Tuple[X, Y, T], float], "Temperature ({.temp_unit})"]
+    """Measured temperature."""
+
+    prec: Ann[Data[Tuple[X, Y, T], float], "Precipitation ({.prec_unit})"]
+    """Measured precipitation."""
+
+    lon: Union[Ann[Coordof[Longitude], "Longitude"], Collection[float]]
+    """Longitude of the measured location."""
+
+    lat: Union[Ann[Coordof[Latitude], "Latitude"], Collection[float]]
+    """Latitude of the measured location."""
+
+    time: Coord[T, L["M8[ns]"]]
+    """Measured time."""
+
+    reference_time: Union[Coord[_, L["M8[ns]"]], np.datetime64]
+    """Reference time."""
+
+    temp_unit: str = "deg C"
+    """Unit of the temperature."""
+
+    prec_unit: str = "mm"
+    """Unit of the precipitation."""
+
+
+weather = Weather(
+    15 + 8 * np.random.randn(2, 2, 3),
+    10 * np.random.rand(2, 2, 3),
+    np.array([[-99.83, -99.32], [-99.79, -99.23]]),
+    np.array([[42.25, 42.21], [42.63, 42.59]]),
+    np.array(["2014-09-06", "2014-09-07", "2014-09-08"], "M8[ns]"),
+    np.datetime64("2014-09-05"),
+)

tests/test_v2_specs.py

@@ -0,0 +1,157 @@
+# standard library
+from dataclasses import MISSING
+
+
+# dependencies
+from test_data import Latitude, Longitude, Weather, weather
+from xarray_dataclasses.v2.specs import Spec
+
+
+# test data
+spec = Spec.from_dataclass(Weather)
+spec_updated = spec @ weather
+
+
+# test functions
+def test_temp() -> None:
+    field = spec.fields.of_data[0]
+
+    assert field.id == "temp"
+    assert field.name == "Temperature ({.temp_unit})"
+    assert field.role == "data"
+    assert field.default is MISSING
+    assert field.type is None
+    assert field.dims == ("x", "y", "time")
+    assert field.dtype == "float64"
+
+
+def test_temp_updated() -> None:
+    field = spec_updated.fields.of_data[0]
+
+    assert field.id == "temp"
+    assert field.name == "Temperature (deg C)"
+    assert field.role == "data"
+    assert (field.default == weather.temp).all()
+    assert field.type is None
+    assert field.dims == ("x", "y", "time")
+    assert field.dtype == "float64"
+
+
+def test_prec() -> None:
+    field = spec.fields.of_data[1]
+
+    assert field.id == "prec"
+    assert field.name == "Precipitation ({.prec_unit})"
+    assert field.role == "data"
+    assert field.default is MISSING
+    assert field.type is None
+    assert field.dims == ("x", "y", "time")
+    assert field.dtype == "float64"
+
+
+def test_prec_updated() -> None:
+    field = spec_updated.fields.of_data[1]
+
+    assert field.id == "prec"
+    assert field.name == "Precipitation (mm)"
+    assert field.role == "data"
+    assert (field.default == weather.prec).all()
+    assert field.type is None
+    assert field.dims == ("x", "y", "time")
+    assert field.dtype == "float64"
+
+
+def test_lon() -> None:
+    field = spec.fields.of_coord[0]
+
+    assert field.id == "lon"
+    assert field.name == "Longitude"
+    assert field.role == "coord"
+    assert field.default is MISSING
+    assert field.type is Longitude
+    assert field.dims == ("x", "y")
+    assert field.dtype == "float64"
+
+
+def test_lon_updated() -> None:
+    field = spec_updated.fields.of_coord[0]
+
+    assert field.id == "lon"
+    assert field.name == "Longitude"
+    assert field.role == "coord"
+    assert (field.default == weather.lon).all()
+    assert field.type is Longitude
+    assert field.dims == ("x", "y")
+    assert field.dtype == "float64"
+
+
+def test_lat() -> None:
+    field = spec.fields.of_coord[1]
+
+    assert field.id == "lat"
+    assert field.name == "Latitude"
+    assert field.role == "coord"
+    assert field.default is MISSING
+    assert field.type is Latitude
+    assert field.dims == ("x", "y")
+    assert field.dtype == "float64"
+
+
+def test_lat_updated() -> None:
+    field = spec_updated.fields.of_coord[1]
+
+    assert field.id == "lat"
+    assert field.name == "Latitude"
+    assert field.role == "coord"
+    assert (field.default == weather.lat).all()
+    assert field.type is Latitude
+    assert field.dims == ("x", "y")
+    assert field.dtype == "float64"
+
+
+def test_time() -> None:
+    field = spec.fields.of_coord[2]
+
+    assert field.id == "time"
+    assert field.name == "time"
+    assert field.role == "coord"
+    assert field.default is MISSING
+    assert field.type is None
+    assert field.dims == ("time",)
+    assert field.dtype == "datetime64[ns]"
+
+
+def test_time_updated() -> None:
+    field = spec_updated.fields.of_coord[2]
+
+    assert field.id == "time"
+    assert field.name == "time"
+    assert field.role == "coord"
+    assert (field.default == weather.time).all()
+    assert field.type is None
+    assert field.dims == ("time",)
+    assert field.dtype == "datetime64[ns]"
+
+
+def test_reference_time() -> None:
+    field = spec.fields.of_coord[3]
+
+    assert field.id == "reference_time"
+    assert field.name == "reference_time"
+    assert field.role == "coord"
+    assert field.default is MISSING
+    assert field.type is None
+    assert field.dims == ()
+    assert field.dtype == "datetime64[ns]"
+
+
+def test_reference_time_updated() -> None:
+    field = spec_updated.fields.of_coord[3]
+
+    assert field.id == "reference_time"
+    assert field.name == "reference_time"
+    assert field.role == "coord"
+    assert field.default == weather.reference_time
+    assert field.type is None
+    assert field.dims == ()
+    assert field.dtype == "datetime64[ns]"