Python only_squaresの例

コード例 #1

ファイル: test_matmul.py プロジェクト: cbm755/diofant

def test_only_squares():
    assert only_squares(C) == [C]
    assert only_squares(C, D) == [C, D]
    assert only_squares(C, A, A.T, D) == [C, A*A.T, D]

    pytest.raises(RuntimeError, lambda: only_squares(C, A))

コード例 #2

def test_only_squares():
    assert only_squares(C) == [C]
    assert only_squares(C, D) == [C, D]
    assert only_squares(C, A, A.T, D) == [C, A * A.T, D]