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
