blob: 629a710e68585d91416c4c3066ca9c5b68c2a491 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
# 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)
|