llvm/mlir/test/python/dialects/sparse_tensor/passes.py

# RUN: %PYTHON %s | FileCheck %s

from mlir.ir import *
from mlir.passmanager import *

from mlir.dialects import sparse_tensor as st


def run(f):
    print("\nTEST:", f.__name__)
    f()
    return f


# CHECK-LABEL: TEST: testSparseTensorPass
@run
def testSparseTensorPass():
    with Context() as context:
        PassManager.parse("any(sparsification)")
        PassManager.parse("any(sparse-tensor-conversion)")
    # CHECK: SUCCESS
    print("SUCCESS")