コード例 #1
0
def bc_dist(expr):
    """ Turn  a*[X, Y] into [a*X, a*Y] """
    factor, mat = expr.as_coeff_mmul()
    if factor != 1 and isinstance(unpack(mat), BlockMatrix):
        B = unpack(mat).blocks
        return BlockMatrix([[factor * B[i, j] for j in range(B.cols)]
                            for i in range(B.rows)])
    return expr
コード例 #2
0
def matrix_of(arg):
    return unpack(arg.as_coeff_mmul()[1])
コード例 #3
0
ファイル: test_strategies.py プロジェクト: diofant/diofant 
def test_unpack():
    assert unpack(Basic(2)) == 2
    assert unpack(Basic(2, 3)) == Basic(2, 3)
コード例 #4
0
ファイル: test_strategies.py プロジェクト: skirpichev/diofant 
def test_unpack():
    assert unpack(Basic(2)) == 2
    assert unpack(Basic(2, 3)) == Basic(2, 3)