def test_unified_dim_sizes(): assert unified_dim_sizes([xr.Variable((), 0)]) == {} assert unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("x", [1])]) == { "x": 1 } assert unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("y", [1, 2])]) == { "x": 1, "y": 2, } assert unified_dim_sizes( [ xr.Variable(("x", "z"), [[1]]), xr.Variable(("y", "z"), [[1, 2], [3, 4]]) ], exclude_dims={"z"}, ) == { "x": 1, "y": 2 } # duplicate dimensions with pytest.raises(ValueError): unified_dim_sizes([xr.Variable(("x", "x"), [[1]])]) # mismatched lengths with pytest.raises(ValueError): unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("x", [1, 2])])
def test_unified_dim_sizes(): assert unified_dim_sizes([xr.Variable((), 0)]) == OrderedDict() assert (unified_dim_sizes([xr.Variable('x', [1]), xr.Variable('x', [1])]) == OrderedDict([('x', 1)])) assert (unified_dim_sizes([xr.Variable('x', [1]), xr.Variable('y', [1, 2])]) == OrderedDict([('x', 1), ('y', 2)])) assert (unified_dim_sizes([xr.Variable(('x', 'z'), [[1]]), xr.Variable(('y', 'z'), [[1, 2], [3, 4]])], exclude_dims={'z'}) == OrderedDict([('x', 1), ('y', 2)])) # duplicate dimensions with pytest.raises(ValueError): unified_dim_sizes([xr.Variable(('x', 'x'), [[1]])]) # mismatched lengths with pytest.raises(ValueError): unified_dim_sizes( [xr.Variable('x', [1]), xr.Variable('x', [1, 2])])
def test_unified_dim_sizes(): assert unified_dim_sizes([xr.Variable((), 0)]) == OrderedDict() assert (unified_dim_sizes([xr.Variable('x', [1]), xr.Variable('x', [1])]) == OrderedDict([('x', 1)])) assert (unified_dim_sizes([xr.Variable('x', [1]), xr.Variable('y', [1, 2])]) == OrderedDict([('x', 1), ('y', 2)])) assert (unified_dim_sizes([xr.Variable(('x', 'z'), [[1]]), xr.Variable(('y', 'z'), [[1, 2], [3, 4]])], exclude_dims={'z'}) == OrderedDict([('x', 1), ('y', 2)])) # duplicate dimensions with pytest.raises(ValueError): unified_dim_sizes([xr.Variable(('x', 'x'), [[1]])]) # mismatched lengths with pytest.raises(ValueError): unified_dim_sizes( [xr.Variable('x', [1]), xr.Variable('x', [1, 2])])
def test_unified_dim_sizes(): assert unified_dim_sizes([xr.Variable((), 0)]) == OrderedDict() assert unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("x", [1])]) == OrderedDict([("x", 1)]) assert unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("y", [1, 2])]) == OrderedDict([("x", 1), ("y", 2)]) assert unified_dim_sizes( [ xr.Variable(("x", "z"), [[1]]), xr.Variable(("y", "z"), [[1, 2], [3, 4]]) ], exclude_dims={"z"}, ) == OrderedDict([("x", 1), ("y", 2)]) # duplicate dimensions with pytest.raises(ValueError): unified_dim_sizes([xr.Variable(("x", "x"), [[1]])]) # mismatched lengths with pytest.raises(ValueError): unified_dim_sizes([xr.Variable("x", [1]), xr.Variable("x", [1, 2])])