llvm/mlir/test/python/ir/debug.py

# RUN: %PYTHON %s | FileCheck %s

from mlir.ir import *


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


# CHECK-LABEL: TEST: testNameIsPrivate
def testNameIsPrivate():
    # `import *` ignores private names starting with an understore, so the debug
    # flag shouldn't be visible unless explicitly imported.
    try:
        _GlobalDebug.flag = True
    except NameError:
        pass
    else:
        assert False, "_GlobalDebug must not be available by default"


run(testNameIsPrivate)


# CHECK-LABEL: TEST: testDebugDlag
def testDebugDlag():
    # Private names must be imported expilcitly.
    from mlir.ir import _GlobalDebug

    # CHECK: False
    print(_GlobalDebug.flag)
    _GlobalDebug.flag = True
    # CHECK: True
    print(_GlobalDebug.flag)
    _GlobalDebug.flag = False
    # CHECK: False
    print(_GlobalDebug.flag)


run(testDebugDlag)