def test_variables__multiple_specs(dummy_ds: xr.Dataset) -> None:
spec = ArrayLikeSpec("baz", "baz doc", kind="i", ndim=1)
invalid_spec = ArrayLikeSpec("baz", "baz doc", kind="i", ndim=2)
variables.validate(dummy_ds, {"foo": spec, "bar": spec})
variables.validate(dummy_ds, {"foo": spec})
variables.validate(dummy_ds, {"bar": spec})
with pytest.raises(ValueError, match="bar does not match the spec"):
variables.validate(dummy_ds, {"bar": invalid_spec})
with pytest.raises(ValueError, match="bar does not match the spec"):
variables.validate(dummy_ds, {"foo": spec}, {"bar": invalid_spec})
def test_variables__whole_ds(dummy_ds: xr.Dataset) -> None:
spec_foo = ArrayLikeSpec("foo", "foo doc", kind="i", ndim=1)
spec_bar = ArrayLikeSpec("bar", "bar doc", kind="i", ndim=1)
try:
SgkitVariables.register_variable(spec_foo)
with pytest.raises(ValueError, match="`foo` already registered"):
SgkitVariables.register_variable(spec_foo)
SgkitVariables.register_variable(spec_bar)
variables.validate(dummy_ds)
finally:
SgkitVariables.registered_variables.pop("foo", None)
SgkitVariables.registered_variables.pop("bar", None)
def test_variables__validate_by_name(dummy_ds: xr.Dataset) -> None:
spec = ArrayLikeSpec("foo", "foo doc", kind="i", ndim=1)
try:
assert "foo" not in SgkitVariables.registered_variables
name, spec_b = SgkitVariables.register_variable(spec)
assert "foo" in SgkitVariables.registered_variables
assert name == "foo"
assert spec_b == spec
variables.validate(dummy_ds, "foo")
finally:
SgkitVariables.registered_variables.pop("foo", None)
assert "foo" not in SgkitVariables.registered_variables
def test_variables_in_multi_index(dummy_ds: xr.Dataset) -> None:
# create a multi index
ds = dummy_ds.set_index({"ind": ("foo", "bar")})
spec = ArrayLikeSpec("foo", "foo doc", kind="i", ndim=1)
variables.validate(ds, spec)
def test_variables__no_present_in_ds(dummy_ds: xr.Dataset) -> None:
spec = ArrayLikeSpec("baz", "baz doc", kind="i", ndim=1)
with pytest.raises(ValueError, match="foobarbaz not present in"):
variables.validate(dummy_ds, {"foobarbaz": spec})
def test_variables__alternative_names(dummy_ds: xr.Dataset) -> None:
spec = ArrayLikeSpec("baz", "baz doc", kind="i", ndim=1)
variables.validate(dummy_ds, {"foo": spec, "bar": spec})
def test_variables__invalid_spec_fails(dummy_ds: xr.Dataset) -> None:
invalid_spec = ArrayLikeSpec("foo", "foo doc", kind="i", ndim=2)
with pytest.raises(ValueError, match="foo does not match the spec"):
variables.validate(dummy_ds, invalid_spec)
def test_variables__validate_by_dummy_spec(dummy_ds: xr.Dataset) -> None:
spec = ArrayLikeSpec("foo", "foo doc", kind="i", ndim=1)
variables.validate(dummy_ds, spec)
