[pypy-svn] r73556 - pypy/branch/cpython-extension/pypy/module/cpyext/test

[jandem at codespeak.net]

Author: jandem
Date: Thu Apr  8 17:36:44 2010
New Revision: 73556

Added:
   pypy/branch/cpython-extension/pypy/module/cpyext/test/test_bufferobject.py
Log:
add tests for bufferobject


Added: pypy/branch/cpython-extension/pypy/module/cpyext/test/test_bufferobject.py
==============================================================================
--- (empty file)
+++ pypy/branch/cpython-extension/pypy/module/cpyext/test/test_bufferobject.py	Thu Apr  8 17:36:44 2010
@@ -0,0 +1,29 @@
+from pypy.rpython.lltypesystem import rffi, lltype
+from pypy.module.cpyext.test.test_api import BaseApiTest
+from pypy.module.cpyext.test.test_cpyext import AppTestCpythonExtensionBase
+from pypy.module.cpyext.api import PyObject
+from pypy.module.cpyext.pyobject import Py_DecRef
+
+class AppTestBufferObject(AppTestCpythonExtensionBase):
+    def test_FromMemory(self):
+        module = self.import_extension('foo', [
+            ("get_FromMemory", "METH_NOARGS",
+             """
+                 cbuf = malloc(4);
+                 cbuf[0] = 'a';
+                 cbuf[1] = 'b';
+                 cbuf[2] = 'c';
+                 cbuf[3] = '\0';
+                 return PyBuffer_FromMemory(cbuf, 4);
+             """),
+            ("free_buffer", "METH_NOARGS",
+             """
+                 free(cbuf);
+                 Py_RETURN_NONE;
+             """)
+            ], prologue = """
+            char* cbuf = NULL;
+            """)
+        w_buffer = module.get_FromMemory()
+        assert False, w_buffer
+        module.free_buffer()