del d.extra
self.assertIsNone(d.extra)
- def test_sys_getobject(self):
- getobject = _testcapi.sys_getobject
-
- self.assertIs(getobject(b'stdout'), sys.stdout)
- with support.swap_attr(sys, '\U0001f40d', 42):
- self.assertEqual(getobject('\U0001f40d'.encode()), 42)
-
- self.assertIs(getobject(b'nonexisting'), AttributeError)
- self.assertIs(getobject(b'\xff'), AttributeError)
- # CRASHES getobject(NULL)
-
- def test_sys_setobject(self):
- setobject = _testcapi.sys_setobject
-
- value = ['value']
- value2 = ['value2']
- try:
- self.assertEqual(setobject(b'newattr', value), 0)
- self.assertIs(sys.newattr, value)
- self.assertEqual(setobject(b'newattr', value2), 0)
- self.assertIs(sys.newattr, value2)
- self.assertEqual(setobject(b'newattr', NULL), 0)
- self.assertFalse(hasattr(sys, 'newattr'))
- self.assertEqual(setobject(b'newattr', NULL), 0)
- finally:
- with contextlib.suppress(AttributeError):
- del sys.newattr
- try:
- self.assertEqual(setobject('\U0001f40d'.encode(), value), 0)
- self.assertIs(getattr(sys, '\U0001f40d'), value)
- self.assertEqual(setobject('\U0001f40d'.encode(), NULL), 0)
- self.assertFalse(hasattr(sys, '\U0001f40d'))
- finally:
- with contextlib.suppress(AttributeError):
- delattr(sys, '\U0001f40d')
-
- with self.assertRaises(UnicodeDecodeError):
- setobject(b'\xff', value)
- # CRASHES setobject(NULL, value)
-
@requires_limited_api
class TestHeapTypeRelative(unittest.TestCase):
--- /dev/null
+import unittest
+import contextlib
+import sys
+from test import support
+from test.support import import_helper
+
+try:
+ import _testcapi
+except ImportError:
+ _testcapi = None
+
+NULL = None
+
+class CAPITest(unittest.TestCase):
+ # TODO: Test the following functions:
+ #
+ # PySys_Audit()
+
+ maxDiff = None
+
+ @support.cpython_only
+ @unittest.skipIf(_testcapi is None, 'need _testcapi module')
+ def test_sys_getobject(self):
+ # Test PySys_GetObject()
+ getobject = _testcapi.sys_getobject
+
+ self.assertIs(getobject(b'stdout'), sys.stdout)
+ with support.swap_attr(sys, '\U0001f40d', 42):
+ self.assertEqual(getobject('\U0001f40d'.encode()), 42)
+
+ self.assertIs(getobject(b'nonexisting'), AttributeError)
+ self.assertIs(getobject(b'\xff'), AttributeError)
+ # CRASHES getobject(NULL)
+
+ @support.cpython_only
+ @unittest.skipIf(_testcapi is None, 'need _testcapi module')
+ def test_sys_setobject(self):
+ # Test PySys_SetObject()
+ setobject = _testcapi.sys_setobject
+
+ value = ['value']
+ value2 = ['value2']
+ try:
+ self.assertEqual(setobject(b'newattr', value), 0)
+ self.assertIs(sys.newattr, value)
+ self.assertEqual(setobject(b'newattr', value2), 0)
+ self.assertIs(sys.newattr, value2)
+ self.assertEqual(setobject(b'newattr', NULL), 0)
+ self.assertFalse(hasattr(sys, 'newattr'))
+ self.assertEqual(setobject(b'newattr', NULL), 0)
+ finally:
+ with contextlib.suppress(AttributeError):
+ del sys.newattr
+ try:
+ self.assertEqual(setobject('\U0001f40d'.encode(), value), 0)
+ self.assertIs(getattr(sys, '\U0001f40d'), value)
+ self.assertEqual(setobject('\U0001f40d'.encode(), NULL), 0)
+ self.assertFalse(hasattr(sys, '\U0001f40d'))
+ finally:
+ with contextlib.suppress(AttributeError):
+ delattr(sys, '\U0001f40d')
+
+ with self.assertRaises(UnicodeDecodeError):
+ setobject(b'\xff', value)
+ # CRASHES setobject(NULL, value)
+
+ @support.cpython_only
+ @unittest.skipIf(_testcapi is None, 'need _testcapi module')
+ def test_sys_getxoptions(self):
+ # Test PySys_GetXOptions()
+ getxoptions = _testcapi.sys_getxoptions
+
+ self.assertIs(getxoptions(), sys._xoptions)
+
+ xoptions = sys._xoptions
+ try:
+ sys._xoptions = 'non-dict'
+ self.assertEqual(getxoptions(), {})
+ self.assertIs(getxoptions(), sys._xoptions)
+
+ del sys._xoptions
+ self.assertEqual(getxoptions(), {})
+ self.assertIs(getxoptions(), sys._xoptions)
+ finally:
+ sys._xoptions = xoptions
+ self.assertIs(getxoptions(), sys._xoptions)
+
+ def _test_sys_formatstream(self, funname, streamname):
+ import_helper.import_module('ctypes')
+ from ctypes import pythonapi, c_char_p, py_object
+ func = getattr(pythonapi, funname)
+ func.argtypes = (c_char_p,)
+
+ # Supports plain C types.
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %s!', c_char_p(b'world'))
+ self.assertEqual(stream.getvalue(), 'Hello, world!')
+
+ # Supports Python objects.
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %R!', py_object('world'))
+ self.assertEqual(stream.getvalue(), "Hello, 'world'!")
+
+ # The total length is not limited.
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %s!', c_char_p(b'world'*200))
+ self.assertEqual(stream.getvalue(), 'Hello, ' + 'world'*200 + '!')
+
+ def test_sys_formatstdout(self):
+ # Test PySys_FormatStdout()
+ self._test_sys_formatstream('PySys_FormatStdout', 'stdout')
+
+ def test_sys_formatstderr(self):
+ # Test PySys_FormatStderr()
+ self._test_sys_formatstream('PySys_FormatStderr', 'stderr')
+
+ def _test_sys_writestream(self, funname, streamname):
+ import_helper.import_module('ctypes')
+ from ctypes import pythonapi, c_char_p
+ func = getattr(pythonapi, funname)
+ func.argtypes = (c_char_p,)
+
+ # Supports plain C types.
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %s!', c_char_p(b'world'))
+ self.assertEqual(stream.getvalue(), 'Hello, world!')
+
+ # There is a limit on the total length.
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %s!', c_char_p(b'world'*100))
+ self.assertEqual(stream.getvalue(), 'Hello, ' + 'world'*100 + '!')
+ with support.captured_output(streamname) as stream:
+ func(b'Hello, %s!', c_char_p(b'world'*200))
+ out = stream.getvalue()
+ self.assertEqual(out[:20], 'Hello, worldworldwor')
+ self.assertEqual(out[-13:], '... truncated')
+ self.assertGreater(len(out), 1000)
+
+ def test_sys_writestdout(self):
+ # Test PySys_WriteStdout()
+ self._test_sys_writestream('PySys_WriteStdout', 'stdout')
+
+ def test_sys_writestderr(self):
+ # Test PySys_WriteStderr()
+ self._test_sys_writestream('PySys_WriteStderr', 'stderr')
+
+
+if __name__ == "__main__":
+ unittest.main()
--- /dev/null
+#define PY_SSIZE_T_CLEAN
+#include "parts.h"
+#include "util.h"
+
+
+static PyObject *
+sys_getobject(PyObject *Py_UNUSED(module), PyObject *arg)
+{
+ const char *name;
+ Py_ssize_t size;
+ if (!PyArg_Parse(arg, "z#", &name, &size)) {
+ return NULL;
+ }
+ PyObject *result = PySys_GetObject(name);
+ if (result == NULL) {
+ result = PyExc_AttributeError;
+ }
+ return Py_NewRef(result);
+}
+
+static PyObject *
+sys_setobject(PyObject *Py_UNUSED(module), PyObject *args)
+{
+ const char *name;
+ Py_ssize_t size;
+ PyObject *value;
+ if (!PyArg_ParseTuple(args, "z#O", &name, &size, &value)) {
+ return NULL;
+ }
+ NULLABLE(value);
+ RETURN_INT(PySys_SetObject(name, value));
+}
+
+static PyObject *
+sys_getxoptions(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(ignored))
+{
+ PyObject *result = PySys_GetXOptions();
+ return Py_XNewRef(result);
+}
+
+
+static PyMethodDef test_methods[] = {
+ {"sys_getobject", sys_getobject, METH_O},
+ {"sys_setobject", sys_setobject, METH_VARARGS},
+ {"sys_getxoptions", sys_getxoptions, METH_NOARGS},
+ {NULL},
+};
+
+int
+_PyTestCapi_Init_Sys(PyObject *m)
+{
+ if (PyModule_AddFunctions(m, test_methods) < 0) {
+ return -1;
+ }
+
+ return 0;
+}
}
-static PyObject *
-sys_getobject(PyObject *Py_UNUSED(module), PyObject *arg)
-{
- const char *name;
- Py_ssize_t size;
- if (!PyArg_Parse(arg, "z#", &name, &size)) {
- return NULL;
- }
- PyObject *result = PySys_GetObject(name);
- if (result == NULL) {
- result = PyExc_AttributeError;
- }
- return Py_NewRef(result);
-}
-
-static PyObject *
-sys_setobject(PyObject *Py_UNUSED(module), PyObject *args)
-{
- const char *name;
- Py_ssize_t size;
- PyObject *value;
- if (!PyArg_ParseTuple(args, "z#O", &name, &size, &value)) {
- return NULL;
- }
- NULLABLE(value);
- RETURN_INT(PySys_SetObject(name, value));
-}
-
-
static PyObject *test_buildvalue_issue38913(PyObject *, PyObject *);
static PyMethodDef TestMethods[] = {
{"function_get_kw_defaults", function_get_kw_defaults, METH_O, NULL},
{"function_set_kw_defaults", function_set_kw_defaults, METH_VARARGS, NULL},
{"test_atexit", test_atexit, METH_NOARGS},
- {"sys_getobject", sys_getobject, METH_O},
- {"sys_setobject", sys_setobject, METH_VARARGS},
{NULL, NULL} /* sentinel */
};
if (_PyTestCapi_Init_PyOS(m) < 0) {
return NULL;
}
+ if (_PyTestCapi_Init_Sys(m) < 0) {
+ return NULL;
+ }
if (_PyTestCapi_Init_Immortal(m) < 0) {
return NULL;
}
projects / thirdparty / Python / cpython.git / commitdiff
