Added EngineBuilder.

Added test/enginebuilder.
This commit is contained in:
Siu Kwan Lam 2012-07-25 15:31:18 -07:00
commit 5bb4b8affb
8 changed files with 267 additions and 3 deletions

56
test/enginebuilder.py Normal file
View file

@ -0,0 +1,56 @@
from llvm.ee import *
from llvm.core import *
from llvm import _util, LLVMException
from ctypes import *
import unittest, logging
class TestEngineBuilder(unittest.TestCase):
def make_test_module(self):
module = Module.new("testmodule")
fnty = Type.function(Type.int(), [])
function = module.add_function(fnty, 'foo')
bb_entry = function.append_basic_block('entry')
builder = Builder.new(bb_entry)
builder.ret(Constant.int(Type.int(), 0xcafe))
module.verify()
return module
def run_foo(self, ee, module):
function = module.get_function_named('foo')
retval = ee.run_function(function, [])
self.assertEqual(retval.as_int(), 0xcafe)
def test_enginebuilder_basic(self):
module = self.make_test_module()
ee = EngineBuilder.new(module).create()
with self.assertRaises(LLVMException):
# Ensure the module is owned.
_util.check_is_unowned(module)
self.run_foo(ee, module)
def test_enginebuilder_force_jit(self):
module = self.make_test_module()
ee = EngineBuilder.new(module).force_jit().create()
self.run_foo(ee, module)
def test_enginebuilder_force_interpreter(self):
module = self.make_test_module()
ee = EngineBuilder.new(module).force_interpreter().create()
self.run_foo(ee, module)
def test_enginebuilder_opt(self):
module = self.make_test_module()
ee = EngineBuilder.new(module).opt(3).create()
self.run_foo(ee, module)
if __name__ == '__main__':
unittest.main()