add object cahce tests

llvm/test_llvmpy.py

@@ -14,7 +14,7 @@ else:
 
 
 from llvm import __version__
-import llvm.core as lc
+from llvm.core import Module, Type, GlobalVariable, Function, Builder
 import llvm.passes as lp
 import llvm.ee as le
 
@@ -39,7 +39,7 @@ entry:
 }
 """
     def test_operands(self):
-        m = lc.Module.from_assembly(StringIO(self.test_module))
+        m = Module.from_assembly(StringIO(self.test_module))
 
         test_func = m.get_function_named("test_func")
         prod = m.get_function_named("prod")
@@ -89,7 +89,7 @@ entry:
 }
 """
     def test_passes(self):
-        m = lc.Module.from_assembly(StringIO(self.asm))
+        m = Module.from_assembly(StringIO(self.asm))
 
         fn_test1 = m.get_function_named('test1')
         fn_test2 = m.get_function_named('test2')
@@ -137,7 +137,7 @@ entry:
         self.assertNotEqual(str(fn_test1).strip(), original_test1.strip())
 
     def test_passes_with_pmb(self):
-        m = lc.Module.from_assembly(StringIO(self.asm))
+        m = Module.from_assembly(StringIO(self.asm))
 
         fn_test1 = m.get_function_named('test1')
         fn_test2 = m.get_function_named('test2')
@@ -189,6 +189,107 @@ tests.append(TestPasses)
 
 # ---------------------------------------------------------------------------
 
+class TestObjCache(unittest.TestCase):
+
+    def test_objcache(self):
+        # Testing module aliasing
+        m1 = Module.new('a')
+        t = Type.int()
+        ft = Type.function(t, [t])
+        f1 = m1.add_function(ft, "func")
+        m2 = f1.module
+        self.assert_(m1 is m2)
+
+        # Testing global vairable aliasing 1
+        gv1 = GlobalVariable.new(m1, t, "gv")
+        gv2 = GlobalVariable.get(m1, "gv")
+        self.assert_(gv1 is gv2)
+
+        # Testing global vairable aliasing 2
+        gv3 = m1.global_variables[0]
+        self.assert_(gv1 is gv3)
+
+        # Testing global vairable aliasing 3
+        gv2 = None
+        gv3 = None
+
+        gv1.delete()
+        gv4 = GlobalVariable.new(m1, t, "gv")
+
+        self.assert_(gv1 is not gv4)
+
+        # Testing function aliasing 1
+        b1 = f1.append_basic_block('entry')
+        f2 = b1.function
+        self.assert_(f1 is f2)
+
+        # Testing function aliasing 2
+        f3 = m1.get_function_named("func")
+        self.assert_(f1 is f3)
+
+        # Testing function aliasing 3
+        f4 = Function.get_or_insert(m1, ft, "func")
+        self.assert_(f1 is f4)
+
+        # Testing function aliasing 4
+        f5 = Function.get(m1, "func")
+        self.assert_(f1 is f5)
+
+        # Testing function aliasing 5
+        f6 = m1.get_or_insert_function(ft, "func")
+        self.assert_(f1 is f6)
+
+        # Testing function aliasing 6
+        f7 = m1.functions[0]
+        self.assert_(f1 is f7)
+
+        # Testing argument aliasing
+        a1 = f1.args[0]
+        a2 = f1.args[0]
+        self.assert_(a1 is a2)
+
+        # Testing basic block aliasing 1
+        b2 = f1.basic_blocks[0]
+        self.assert_(b1 is b2)
+
+        # Testing basic block aliasing 2
+        b3 = f1.get_entry_basic_block()
+        self.assert_(b1 is b3)
+
+        # Testing basic block aliasing 3
+        b31 = f1.entry_basic_block
+        self.assert_(b1 is b31)
+
+        # Testing basic block aliasing 4
+        bldr = Builder.new(b1)
+        b4 = bldr.basic_block
+        self.assert_(b1 is b4)
+
+        # Testing basic block aliasing 5
+        i1 = bldr.ret_void()
+        b5 = i1.basic_block
+        self.assert_(b1 is b5)
+
+        # Testing instruction aliasing 1
+        i2 = b5.instructions[0]
+        self.assert_(i1 is i2)
+
+        # phi node
+        phi = bldr.phi(t)
+        phi.add_incoming(f1.args[0], b1)
+        v2 = phi.get_incoming_value(0)
+        b6 = phi.get_incoming_block(0)
+
+        # Testing PHI / basic block aliasing 5
+        self.assert_(b1 is b6)
+
+        # Testing PHI / value aliasing
+        self.assert_(f1.args[0] is v2)
+
+tests.append(TestObjCache)
+
+# ---------------------------------------------------------------------------
+
 def run(verbosity=1):
     print('llvmpy is installed in: ' + os.path.dirname(__file__))
     print('llvmpy version: ' + __version__)