summaryrefslogtreecommitdiff
path: root/bindings/python/llvm/tests/test_core.py
diff options
context:
space:
mode:
Diffstat (limited to 'bindings/python/llvm/tests/test_core.py')
-rw-r--r--bindings/python/llvm/tests/test_core.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/bindings/python/llvm/tests/test_core.py b/bindings/python/llvm/tests/test_core.py
index a1f79a490d..67e294b056 100644
--- a/bindings/python/llvm/tests/test_core.py
+++ b/bindings/python/llvm/tests/test_core.py
@@ -78,3 +78,25 @@ class TestCore(TestBase):
self.assertEqual(f.name, functions[i])
f.dump()
+ def test_function_basicblock_iteration(self):
+ m = parse_bitcode(MemoryBuffer(filename=self.get_test_bc()))
+ i = 0
+
+ bb_list = ['b1', 'b2', 'end']
+
+ f = m.first
+ while f.name != "f6":
+ f = f.next
+
+ # Forward
+ for bb in f:
+ self.assertEqual(bb.name, bb_list[i])
+ bb.dump()
+ i += 1
+
+ # Backwards
+ for bb in reversed(f):
+ i -= 1
+ self.assertEqual(bb.name, bb_list[i])
+ bb.dump()
+