Python block示例

示例#1

文件： test_eigen.py 项目： phaustin/pybind11

def test_nonunit_stride_to_python():
    from pybind11_tests import diagonal, diagonal_1, diagonal_n, block

    assert np.all(diagonal(ref) == ref.diagonal())
    assert np.all(diagonal_1(ref) == ref.diagonal(1))
    for i in range(-5, 7):
        assert np.all(diagonal_n(ref, i) == ref.diagonal(i)), "diagonal_n({})".format(i)

    assert np.all(block(ref, 2, 1, 3, 3) == ref[2:5, 1:4])
    assert np.all(block(ref, 1, 4, 4, 2) == ref[1:, 4:])
    assert np.all(block(ref, 1, 4, 3, 2) == ref[1:4, 4:])

示例#2

文件： test_eigen.py 项目： jagerman/pybind11

def test_nonunit_stride_to_python():
    from pybind11_tests import diagonal, diagonal_1, diagonal_n, block

    assert np.all(diagonal(ref) == ref.diagonal())
    assert np.all(diagonal_1(ref) == ref.diagonal(1))
    for i in range(-5, 7):
        assert np.all(diagonal_n(ref, i) == ref.diagonal(i)), "diagonal_n({})".format(i)

    assert np.all(block(ref, 2, 1, 3, 3) == ref[2:5, 1:4])
    assert np.all(block(ref, 1, 4, 4, 2) == ref[1:, 4:])
    assert np.all(block(ref, 1, 4, 3, 2) == ref[1:4, 4:])

示例#3

文件： test_eigen.py 项目： dlminvestments/coremltools

def test_nonunit_stride_to_python():
    from pybind11_tests import diagonal, diagonal_1, diagonal_n, block

    if not np.all(diagonal(ref) == ref.diagonal()):
        raise AssertionError
    if not np.all(diagonal_1(ref) == ref.diagonal(1)):
        raise AssertionError
    for i in range(-5, 7):
        if not np.all(diagonal_n(ref, i) == ref.diagonal(i)):
            raise AssertionError("diagonal_n({})".format(i))

    if not np.all(block(ref, 2, 1, 3, 3) == ref[2:5, 1:4]):
        raise AssertionError
    if not np.all(block(ref, 1, 4, 4, 2) == ref[1:, 4:]):
        raise AssertionError
    if not np.all(block(ref, 1, 4, 3, 2) == ref[1:4, 4:]):
        raise AssertionError