self.assertEqual(co.co_name, "funcname")
self.assertEqual(co.co_firstlineno, 15)
+
+def isinterned(s):
+ return s is intern(('_' + s + '_')[1:-1])
+
class CodeConstsTest(unittest.TestCase):
def find_const(self, consts, value):
for v in consts:
if v == value:
return v
- self.assertIn(value, consts) # rises an exception
- self.fail('Should be never reached')
+ self.assertIn(value, consts) # raises an exception
+ self.fail('Should never be reached')
def assertIsInterned(self, s):
- if s is not intern(s):
+ if not isinterned(s):
self.fail('String %r is not interned' % (s,))
+ def assertIsNotInterned(self, s):
+ if isinterned(s):
+ self.fail('String %r is interned' % (s,))
+
@cpython_only
def test_interned_string(self):
co = compile('res = "str_value"', '?', 'exec')
return a
self.assertIsInterned(f())
+ @cpython_only
+ def test_interned_string_with_null(self):
+ co = compile(r'res = "str\0value!"', '?', 'exec')
+ v = self.find_const(co.co_consts, 'str\0value!')
+ self.assertIsNotInterned(v)
+
class CodeWeakRefTest(unittest.TestCase):
/* all_name_chars(s): true iff all chars in s are valid NAME_CHARS */
static int
-all_name_chars(unsigned char *s)
+all_name_chars(PyObject *o)
{
static char ok_name_char[256];
- static unsigned char *name_chars = (unsigned char *)NAME_CHARS;
+ static const unsigned char *name_chars = (unsigned char *)NAME_CHARS;
+ const unsigned char *s, *e;
if (ok_name_char[*name_chars] == 0) {
- unsigned char *p;
+ const unsigned char *p;
for (p = name_chars; *p; p++)
ok_name_char[*p] = 1;
}
- while (*s) {
+ s = (unsigned char *)PyString_AS_STRING(o);
+ e = s + PyString_GET_SIZE(o);
+ while (s != e) {
if (ok_name_char[*s++] == 0)
return 0;
}
for (i = PyTuple_GET_SIZE(tuple); --i >= 0; ) {
PyObject *v = PyTuple_GET_ITEM(tuple, i);
if (PyString_CheckExact(v)) {
- if (all_name_chars((unsigned char *)PyString_AS_STRING(v))) {
+ if (all_name_chars(v)) {
PyObject *w = v;
PyString_InternInPlace(&v);
if (w != v) {