forked from M-Labs/artiq
34 lines
509 B
Python
34 lines
509 B
Python
|
# RUN: %python -m artiq.compiler.testbench.jit %s
|
||
|
|
# RUN: %python %s
|
||
|
|
|
||
|
|
class contextmgr:
|
||
|
|
def __enter__(self):
|
||
|
|
print(2)
|
||
|
|
|
||
|
|
def __exit__(self, n1, n2, n3):
|
||
|
|
print(4)
|
||
|
|
|
||
|
|
# CHECK-L: a 1
|
||
|
|
# CHECK-L: 2
|
||
|
|
# CHECK-L: a 3
|
||
|
|
# CHECK-L: 4
|
||
|
|
# CHECK-L: a 5
|
||
|
|
print("a", 1)
|
||
|
|
with contextmgr():
|
||
|
|
print("a", 3)
|
||
|
|
print("a", 5)
|
||
|
|
|
||
|
|
# CHECK-L: b 1
|
||
|
|
# CHECK-L: 2
|
||
|
|
# CHECK-L: 4
|
||
|
|
# CHECK-L: b 6
|
||
|
|
try:
|
||
|
|
print("b", 1)
|
||
|
|
with contextmgr():
|
||
|
|
[0][1]
|
||
|
|
print("b", 3)
|
||
|
|
print("b", 5)
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
print("b", 6)
|