def test_array_repr_variable(self) -> None: var = xr.Variable("x", [0, 1]) formatting.array_repr(var) with xr.set_options(display_expand_data=False): formatting.array_repr(var)
def test_array_repr(self): ds = xr.Dataset(coords={'foo': [1, 2, 3], 'bar': [1, 2, 3]}) ds[(1, 2)] = xr.DataArray([0], dims='test') actual = formatting.array_repr(ds[(1, 2)]) expected = dedent("""\ <xarray.DataArray (1, 2) (test: 1)> array([0]) Dimensions without coordinates: test""") assert actual == expected
def test_array_repr(self): ds = xr.Dataset(coords={"foo": [1, 2, 3], "bar": [1, 2, 3]}) ds[(1, 2)] = xr.DataArray([0], dims="test") actual = formatting.array_repr(ds[(1, 2)]) expected = dedent("""\ <xarray.DataArray (1, 2) (test: 1)> array([0]) Dimensions without coordinates: test""") assert actual == expected with xr.set_options(display_expand_data=False): actual = formatting.array_repr(ds[(1, 2)]) expected = dedent("""\ <xarray.DataArray (1, 2) (test: 1)> 0 Dimensions without coordinates: test""") assert actual == expected
def test_array_repr(self): ds = xr.Dataset(coords={'foo': [1, 2, 3], 'bar': [1, 2, 3]}) ds[(1, 2)] = xr.DataArray([0], dims='test') actual = formatting.array_repr(ds[(1, 2)]) expected = dedent("""\ <xarray.DataArray (1, 2) (test: 1)> array([0]) Dimensions without coordinates: test""") assert actual == expected
def test_repr_file_collapsed(tmp_path) -> None: arr = xr.DataArray(np.arange(300), dims="test") arr.to_netcdf(tmp_path / "test.nc", engine="netcdf4") with xr.open_dataarray(tmp_path / "test.nc") as arr, xr.set_options( display_expand_data=False): actual = formatting.array_repr(arr) expected = dedent("""\ <xarray.DataArray (test: 300)> array([ 0, 1, 2, ..., 297, 298, 299]) Dimensions without coordinates: test""") assert actual == expected