From abaeee678f3335d3d8dc651acff4646814724e7a Mon Sep 17 00:00:00 2001
From: Ilan Schnell <ilan@continuum.io>
Date: Sat, 22 Sep 2012 20:39:00 -0500
Subject: [PATCH] add volatile test

---
 llvm/test_llvmpy.py | 54 ++++++++++++++++++++++++++++++++++++++++++++
 test/volatile.py    | 55 ---------------------------------------------
 2 files changed, 54 insertions(+), 55 deletions(-)
 delete mode 100644 test/volatile.py

diff --git a/llvm/test_llvmpy.py b/llvm/test_llvmpy.py
index 7a9c3b6..c5a3c69 100644
--- a/llvm/test_llvmpy.py
+++ b/llvm/test_llvmpy.py
@@ -732,6 +732,60 @@ tests.append(TestIntrinsic)
 
 # ---------------------------------------------------------------------------
 
+class TestVolatile(unittest.TestCase):
+
+    def test_volatile(self):
+        mod = Module.new('mod')
+        functype = Type.function(Type.void(), [])
+        func = mod.add_function(functype, name='foo')
+        bb = func.append_basic_block('entry')
+        bldr = Builder.new(bb)
+        ptr = bldr.alloca(Type.int())
+
+        # test load inst
+        val = bldr.load(ptr)
+        self.assertFalse(val.is_volatile, "default must be non-volatile")
+        val.set_volatile(True)
+        self.assertTrue(val.is_volatile, "fail to set volatile")
+        val.set_volatile(False)
+        self.assertFalse(val.is_volatile, "fail to unset volatile")
+
+        # test store inst
+        store_inst = bldr.store(val, ptr)
+        self.assertFalse(store_inst.is_volatile, "default must be non-volatile")
+        store_inst.set_volatile(True)
+        self.assertTrue(store_inst.is_volatile, "fail to set volatile")
+        store_inst.set_volatile(False)
+        self.assertFalse(store_inst.is_volatile, "fail to unset volatile")
+
+    def test_volatile_another(self):
+        mod = Module.new('mod')
+        functype = Type.function(Type.void(), [])
+        func = mod.add_function(functype, name='foo')
+        bb = func.append_basic_block('entry')
+        bldr = Builder.new(bb)
+        ptr = bldr.alloca(Type.int())
+
+        # test load inst
+        val = bldr.load(ptr, volatile=True)
+        self.assertTrue(val.is_volatile, "volatile kwarg does not work")
+        val.set_volatile(False)
+        self.assertFalse(val.is_volatile, "fail to unset volatile")
+        val.set_volatile(True)
+        self.assertTrue(val.is_volatile, "fail to set volatile")
+
+        # test store inst
+        store_inst = bldr.store(val, ptr, volatile=True)
+        self.assertTrue(store_inst.is_volatile, "volatile kwarg does not work")
+        store_inst.set_volatile(False)
+        self.assertFalse(store_inst.is_volatile, "fail to unset volatile")
+        store_inst.set_volatile(True)
+        self.assertTrue(store_inst.is_volatile, "fail to set volatile")
+
+tests.append(TestVolatile)
+
+# ---------------------------------------------------------------------------
+
 def run(verbosity=1):
     print('llvmpy is installed in: ' + os.path.dirname(__file__))
     print('llvmpy version: ' + __version__)
diff --git a/test/volatile.py b/test/volatile.py
deleted file mode 100644
index e567a80..0000000
--- a/test/volatile.py
+++ /dev/null
@@ -1,55 +0,0 @@
-from llvm.core import *
-import unittest
-
-class TestVolatile(unittest.TestCase):
-    def test_volatile(self):
-        mod = Module.new('mod')
-        functype = Type.function(Type.void(), [])
-        func = mod.add_function(functype, name='foo')
-        bb = func.append_basic_block('entry')
-        bldr = Builder.new(bb)
-        ptr = bldr.alloca(Type.int())
-
-        # test load inst
-        val = bldr.load(ptr)
-        self.assertFalse(val.is_volatile, "default must be non-volatile")
-        val.set_volatile(True)
-        self.assertTrue(val.is_volatile, "fail to set volatile")
-        val.set_volatile(False)
-        self.assertFalse(val.is_volatile, "fail to unset volatile")
-
-        # test store inst
-        store_inst = bldr.store(val, ptr)
-        self.assertFalse(store_inst.is_volatile, "default must be non-volatile")
-        store_inst.set_volatile(True)
-        self.assertTrue(store_inst.is_volatile, "fail to set volatile")
-        store_inst.set_volatile(False)
-        self.assertFalse(store_inst.is_volatile, "fail to unset volatile")
-
-    def test_volatile_another(self):
-        mod = Module.new('mod')
-        functype = Type.function(Type.void(), [])
-        func = mod.add_function(functype, name='foo')
-        bb = func.append_basic_block('entry')
-        bldr = Builder.new(bb)
-        ptr = bldr.alloca(Type.int())
-
-        # test load inst
-        val = bldr.load(ptr, volatile=True)
-        self.assertTrue(val.is_volatile, "volatile kwarg does not work")
-        val.set_volatile(False)
-        self.assertFalse(val.is_volatile, "fail to unset volatile")
-        val.set_volatile(True)
-        self.assertTrue(val.is_volatile, "fail to set volatile")
-
-        # test store inst
-        store_inst = bldr.store(val, ptr, volatile=True)
-        self.assertTrue(store_inst.is_volatile, "volatile kwarg does not work")
-        store_inst.set_volatile(False)
-        self.assertFalse(store_inst.is_volatile, "fail to unset volatile")
-        store_inst.set_volatile(True)
-        self.assertTrue(store_inst.is_volatile, "fail to set volatile")
-
-
-if __name__ == '__main__':
-    unittest.main()