aboutsummaryrefslogtreecommitdiff
path: root/mlir/test/python/ir/operation.py
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/test/python/ir/operation.py')
-rw-r--r--mlir/test/python/ir/operation.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/mlir/test/python/ir/operation.py b/mlir/test/python/ir/operation.py
index ede1571..c6b5daf 100644
--- a/mlir/test/python/ir/operation.py
+++ b/mlir/test/python/ir/operation.py
@@ -978,8 +978,12 @@ def testModuleMerge():
foo = m1.body.operations[0]
bar = m2.body.operations[0]
qux = m2.body.operations[1]
+ assert bar.is_before_in_block(qux)
bar.move_before(foo)
+ assert bar.is_before_in_block(foo)
qux.move_after(foo)
+ assert bar.is_before_in_block(qux)
+ assert foo.is_before_in_block(qux)
# CHECK: module
# CHECK: func private @bar