def test_broadcast_compat_data_2d(): data = np.arange(12).reshape(3, 4) var = xr.Variable(['x', 'y'], data) assert_identical(data, broadcast_compat_data(var, ('x', 'y'), ())) assert_identical(data, broadcast_compat_data(var, ('x', ), ('y', ))) assert_identical(data, broadcast_compat_data(var, (), ('x', 'y'))) assert_identical(data.T, broadcast_compat_data(var, ('y', 'x'), ())) assert_identical(data.T, broadcast_compat_data(var, ('y', ), ('x', ))) assert_identical(data, broadcast_compat_data(var, ('w', 'x'), ('y', ))) assert_identical(data, broadcast_compat_data(var, ('w', ), ('x', 'y'))) assert_identical(data.T, broadcast_compat_data(var, ('w', ), ('y', 'x'))) assert_identical(data[:, :, None], broadcast_compat_data(var, ('w', 'x', 'y', 'z'), ())) assert_identical(data[None, :, :].T, broadcast_compat_data(var, ('w', 'y', 'x', 'z'), ()))
def test_broadcast_compat_data_2d(): data = np.arange(12).reshape(3, 4) var = xr.Variable(["x", "y"], data) assert_identical(data, broadcast_compat_data(var, ("x", "y"), ())) assert_identical(data, broadcast_compat_data(var, ("x", ), ("y", ))) assert_identical(data, broadcast_compat_data(var, (), ("x", "y"))) assert_identical(data.T, broadcast_compat_data(var, ("y", "x"), ())) assert_identical(data.T, broadcast_compat_data(var, ("y", ), ("x", ))) assert_identical(data, broadcast_compat_data(var, ("w", "x"), ("y", ))) assert_identical(data, broadcast_compat_data(var, ("w", ), ("x", "y"))) assert_identical(data.T, broadcast_compat_data(var, ("w", ), ("y", "x"))) assert_identical(data[:, :, None], broadcast_compat_data(var, ("w", "x", "y", "z"), ())) assert_identical(data[None, :, :].T, broadcast_compat_data(var, ("w", "y", "x", "z"), ()))
def test_broadcast_compat_data_2d(): data = np.arange(12).reshape(3, 4) var = xr.Variable(['x', 'y'], data) assert_identical(data, broadcast_compat_data(var, ('x', 'y'), ())) assert_identical(data, broadcast_compat_data(var, ('x',), ('y',))) assert_identical(data, broadcast_compat_data(var, (), ('x', 'y'))) assert_identical(data.T, broadcast_compat_data(var, ('y', 'x'), ())) assert_identical(data.T, broadcast_compat_data(var, ('y',), ('x',))) assert_identical(data, broadcast_compat_data(var, ('w', 'x'), ('y',))) assert_identical(data, broadcast_compat_data(var, ('w',), ('x', 'y'))) assert_identical(data.T, broadcast_compat_data(var, ('w',), ('y', 'x'))) assert_identical(data[:, :, None], broadcast_compat_data(var, ('w', 'x', 'y', 'z'), ())) assert_identical(data[None, :, :].T, broadcast_compat_data(var, ('w', 'y', 'x', 'z'), ()))
def test_broadcast_compat_data_1d(): data = np.arange(5) var = xr.Variable("x", data) assert_identical(data, broadcast_compat_data(var, ("x",), ())) assert_identical(data, broadcast_compat_data(var, (), ("x",))) assert_identical(data[:], broadcast_compat_data(var, ("w",), ("x",))) assert_identical(data[:, None], broadcast_compat_data(var, ("w", "x", "y"), ())) with pytest.raises(ValueError): broadcast_compat_data(var, ("x",), ("w",)) with pytest.raises(ValueError): broadcast_compat_data(var, (), ())
def test_broadcast_compat_data_1d(): data = np.arange(5) var = xr.Variable('x', data) assert_identical(data, broadcast_compat_data(var, ('x',), ())) assert_identical(data, broadcast_compat_data(var, (), ('x',))) assert_identical(data[:], broadcast_compat_data(var, ('w',), ('x',))) assert_identical(data[:, None], broadcast_compat_data(var, ('w', 'x', 'y'), ())) with pytest.raises(ValueError): broadcast_compat_data(var, ('x',), ('w',)) with pytest.raises(ValueError): broadcast_compat_data(var, (), ())