diff --git a/llvm/test_llvmpy.py b/llvm/test_llvmpy.py index 0673efb..4df1b75 100644 --- a/llvm/test_llvmpy.py +++ b/llvm/test_llvmpy.py @@ -71,6 +71,28 @@ tests.append(TestAsm) # --------------------------------------------------------------------------- +class TestAttr(unittest.TestCase): + def make_module(self): + test_module = """ + define i32 @sum(i32, i32) { + entry: + %2 = add i32 %0, %1 + ret i32 %2 + } + """ + return Module.from_assembly(StringIO(test_module)) + + def test_align(self): + m = self.make_module() + f = m.get_function_named('sum') + f.args[0].alignment = 16 + self.assert_("align 16" in str(f)) + self.assertEqual(f.args[0].alignment, 16) + +tests.append(TestAttr) + +# --------------------------------------------------------------------------- + class TestOperands(unittest.TestCase): # implement a test function test_module = """ @@ -575,9 +597,9 @@ class TestIntrinsic(unittest.TestCase): # define float @mysin(float %x) { # entry: # %cosx = call float @llvm.cos.f32( float %x ) ; [#uses -# %sin = call float @llvm.sqrt.f32( float %onemc2 ) +# %sin = call float @llvm.sqrt.f32( float %onemc2 ) # ret float %sin # } # diff --git a/test/testattrs.py b/test/testattrs.py deleted file mode 100644 index 8801f62..0000000 --- a/test/testattrs.py +++ /dev/null @@ -1,31 +0,0 @@ -#!/usr/bin/env python - -from llvm.core import * -try: - from StringIO import StringIO -except ImportError: - from io import StringIO - - -import unittest - -def make_module(): - test_module = """ - define i32 @sum(i32, i32) { - entry: - %2 = add i32 %0, %1 - ret i32 %2 - } - """ - return Module.from_assembly(StringIO(test_module)) - -class TestAttr(unittest.TestCase): - def test_align(self): - m = make_module() - f = m.get_function_named('sum') - f.args[0].alignment = 16 - self.assertIn("align 16", str(f)) - self.assertEqual(f.args[0].alignment, 16) - -if __name__ == '__main__': - unittest.main()