Python blockの例

コード例 #1

def test_block_diag_square_check():
    rows = [
        [generate("diag:6"),
         generate("zero:6,7"),
         generate("zero:6,3")],
        [generate("zero:6,6"),
         generate("zero:6,7"),
         generate("zero:6,3")],
        [generate("zero:3,6"),
         generate("zero:3,7"),
         generate("diag:3")],
    ]
    with AssertDenseWarning("could not preserve structure"):
        assert isinstance(B.block(*rows), Dense)

    rows = [
        [generate("diag:6"),
         generate("zero:6,6"),
         generate("zero:6,6")],
        [generate("zero:6,6"),
         generate("zero:6,6"),
         generate("zero:6,6")],
    ]
    with AssertDenseWarning("could not preserve structure"):
        assert isinstance(B.block(*rows), Dense)

コード例 #2

def test_block_diag():
    rows = [
        [generate("diag:6"),
         generate("zero:6,6"),
         generate("zero:6,3")],
        [generate("zero:6,6"),
         generate("zero:6,6"),
         generate("zero:6,3")],
        [generate("zero:3,6"),
         generate("zero:3,6"),
         generate("diag:3")],
    ]
    res = B.block(*rows)
    approx(res, B.concat2d(*_dense(rows)))
    assert isinstance(res, Diagonal)

コード例 #3

def test_block_zero():
    rows = [[generate("zero:6,6") for _ in range(3)] for _ in range(3)]
    res = B.block(*rows)
    approx(res, B.concat2d(*_dense(rows)))
    assert isinstance(res, Zero)

コード例 #4

def test_block_one_block():
    rows = [[generate("dense:3,3")]]
    assert B.block(*rows) is rows[0][0]

コード例 #5

def test_block_dense():
    rows = [[generate("dense:6,6") for _ in range(3)] for _ in range(3)]
    with AssertDenseWarning("could not preserve structure"):
        res = B.block(*rows)
    approx(res, B.concat2d(*_dense(rows)))

コード例 #6

ファイル: input.py プロジェクト: wesselb/stheno

def pairwise(k, x: tuple, y: tuple):
    return B.block(*[[pairwise(k, xi, yi) for yi in y] for xi in x])