1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from pypy.module.cpyext.test.test_cpyext import AppTestCpythonExtensionBase
class AppTestMarshal(AppTestCpythonExtensionBase):
def test_PyMarshal_ReadObjectFromString(self):
module = self.import_extension('foo', [
("mloads", "METH_O",
"""
char *input = PyBytes_AsString(args);
Py_ssize_t length = PyBytes_Size(args);
return PyMarshal_ReadObjectFromString(input, length);
""")],
prologue='#include <marshal.h>')
import marshal
assert module.mloads(marshal.dumps(42.5)) == 42.5
x = [None, True, (4, 5), b"adkj", u"\u1234"]
assert module.mloads(marshal.dumps(x)) == x
def test_PyMarshal_WriteObjectToString(self):
module = self.import_extension('foo', [
("mdumps", "METH_VARARGS",
"""
PyObject *obj;
int version;
if (!PyArg_ParseTuple(args, "Oi", &obj, &version))
return NULL;
return PyMarshal_WriteObjectToString(obj, version);
""")],
prologue='#include <marshal.h>')
import marshal
for x in [42, b"foo", u"\u2345", (4, None, False)]:
for version in [0, 1, 2]:
assert module.mdumps(x, version) == marshal.dumps(x, version)
|
