Function create_nan_array

xarray/tests/test_units.py:33–42

Create array with NaN values, handling cast warnings for int dtypes.

(values, dtype)

def create_nan_array(values, dtype):
    """Create array with NaN values, handling cast warnings for int dtypes."""
    import warnings

    # When casting float arrays with NaN to integer, NumPy raises a warning
    # This is expected behavior when dtype is int
    with warnings.catch_warnings():
        if np.issubdtype(dtype, np.integer):
            warnings.filterwarnings("ignore", "invalid value encountered in cast")
        return np.array(values).astype(dtype)


# make sure scalars are converted to 0d arrays so quantities can