Python unified_dim_sizes примеры использования

Пример #1

Файл: test_computation.py Проект: jenssss/xarray

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])])

Пример #2

Файл: test_computation.py Проект: tfurf/xarray

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])])

Пример #3

Файл: test_computation.py Проект: edoddridge/xarray

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])])

Пример #4

Файл: test_computation.py Проект: shanaxel42/xarray

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])])