Function test_fill_value_coder_complex

xarray/tests/test_backends.py:7246–7253

Test that FillValueCoder round-trips complex fill values.

(dtype)

@requires_zarr
@pytest.mark.parametrize("dtype", [complex, np.complex64, np.complex128])
def test_fill_value_coder_complex(dtype) -> None:
    """Test that FillValueCoder round-trips complex fill values."""
    from xarray.backends.zarr import FillValueCoder

    for value in [dtype(1 + 2j), dtype(-3.5 + 4.5j), dtype(complex("nan+nanj"))]:
        encoded = FillValueCoder.encode(value, np.dtype(dtype))
        decoded = FillValueCoder.decode(encoded, np.dtype(dtype))
        np.testing.assert_equal(np.array(decoded, dtype=dtype), np.array(value))


@requires_zarr