mirror of https://github.com/python/cpython
823 lines
31 KiB
Python
823 lines
31 KiB
Python
import unittest
|
|
from collections import OrderedDict
|
|
import _testcapi
|
|
|
|
|
|
NULL = None
|
|
|
|
class TestObject:
|
|
@property
|
|
def evil(self):
|
|
raise RuntimeError('do not get evil')
|
|
@evil.setter
|
|
def evil(self, value):
|
|
raise RuntimeError('do not set evil')
|
|
@evil.deleter
|
|
def evil(self):
|
|
raise RuntimeError('do not del evil')
|
|
|
|
class ProxyGetItem:
|
|
def __init__(self, obj):
|
|
self.obj = obj
|
|
def __getitem__(self, key):
|
|
return self.obj[key]
|
|
|
|
class ProxySetItem:
|
|
def __init__(self, obj):
|
|
self.obj = obj
|
|
def __setitem__(self, key, value):
|
|
self.obj[key] = value
|
|
|
|
class ProxyDelItem:
|
|
def __init__(self, obj):
|
|
self.obj = obj
|
|
def __delitem__(self, key):
|
|
del self.obj[key]
|
|
|
|
def gen():
|
|
yield 'a'
|
|
yield 'b'
|
|
yield 'c'
|
|
|
|
|
|
class CAPITest(unittest.TestCase):
|
|
|
|
def test_object_getattr(self):
|
|
xgetattr = _testcapi.object_getattr
|
|
obj = TestObject()
|
|
obj.a = 11
|
|
setattr(obj, '\U0001f40d', 22)
|
|
self.assertEqual(xgetattr(obj, 'a'), 11)
|
|
self.assertRaises(AttributeError, xgetattr, obj, 'b')
|
|
self.assertEqual(xgetattr(obj, '\U0001f40d'), 22)
|
|
|
|
self.assertRaises(RuntimeError, xgetattr, obj, 'evil')
|
|
self.assertRaises(TypeError, xgetattr, obj, 1)
|
|
# CRASHES xgetattr(obj, NULL)
|
|
# CRASHES xgetattr(NULL, 'a')
|
|
|
|
def test_object_getattrstring(self):
|
|
getattrstring = _testcapi.object_getattrstring
|
|
obj = TestObject()
|
|
obj.a = 11
|
|
setattr(obj, '\U0001f40d', 22)
|
|
self.assertEqual(getattrstring(obj, b'a'), 11)
|
|
self.assertRaises(AttributeError, getattrstring, obj, b'b')
|
|
self.assertEqual(getattrstring(obj, '\U0001f40d'.encode()), 22)
|
|
|
|
self.assertRaises(RuntimeError, getattrstring, obj, b'evil')
|
|
self.assertRaises(UnicodeDecodeError, getattrstring, obj, b'\xff')
|
|
# CRASHES getattrstring(obj, NULL)
|
|
# CRASHES getattrstring(NULL, b'a')
|
|
|
|
def test_object_getoptionalattr(self):
|
|
getoptionalattr = _testcapi.object_getoptionalattr
|
|
obj = TestObject()
|
|
obj.a = 11
|
|
setattr(obj, '\U0001f40d', 22)
|
|
self.assertEqual(getoptionalattr(obj, 'a'), 11)
|
|
self.assertIs(getoptionalattr(obj, 'b'), AttributeError)
|
|
self.assertEqual(getoptionalattr(obj, '\U0001f40d'), 22)
|
|
|
|
self.assertRaises(RuntimeError, getoptionalattr, obj, 'evil')
|
|
self.assertRaises(TypeError, getoptionalattr, obj, 1)
|
|
# CRASHES getoptionalattr(obj, NULL)
|
|
# CRASHES getoptionalattr(NULL, 'a')
|
|
|
|
def test_object_getoptionalattrstring(self):
|
|
getoptionalattrstring = _testcapi.object_getoptionalattrstring
|
|
obj = TestObject()
|
|
obj.a = 11
|
|
setattr(obj, '\U0001f40d', 22)
|
|
self.assertEqual(getoptionalattrstring(obj, b'a'), 11)
|
|
self.assertIs(getoptionalattrstring(obj, b'b'), AttributeError)
|
|
self.assertEqual(getoptionalattrstring(obj, '\U0001f40d'.encode()), 22)
|
|
|
|
self.assertRaises(RuntimeError, getoptionalattrstring, obj, b'evil')
|
|
self.assertRaises(UnicodeDecodeError, getoptionalattrstring, obj, b'\xff')
|
|
# CRASHES getoptionalattrstring(obj, NULL)
|
|
# CRASHES getoptionalattrstring(NULL, b'a')
|
|
|
|
def test_object_hasattr(self):
|
|
xhasattr = _testcapi.object_hasattr
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
self.assertTrue(xhasattr(obj, 'a'))
|
|
self.assertFalse(xhasattr(obj, 'b'))
|
|
self.assertTrue(xhasattr(obj, '\U0001f40d'))
|
|
|
|
self.assertFalse(xhasattr(obj, 'evil'))
|
|
self.assertFalse(xhasattr(obj, 1))
|
|
# CRASHES xhasattr(obj, NULL)
|
|
# CRASHES xhasattr(NULL, 'a')
|
|
|
|
def test_object_hasattrstring(self):
|
|
hasattrstring = _testcapi.object_hasattrstring
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
self.assertTrue(hasattrstring(obj, b'a'))
|
|
self.assertFalse(hasattrstring(obj, b'b'))
|
|
self.assertTrue(hasattrstring(obj, '\U0001f40d'.encode()))
|
|
|
|
self.assertFalse(hasattrstring(obj, b'evil'))
|
|
self.assertFalse(hasattrstring(obj, b'\xff'))
|
|
# CRASHES hasattrstring(obj, NULL)
|
|
# CRASHES hasattrstring(NULL, b'a')
|
|
|
|
def test_object_hasattrwitherror(self):
|
|
xhasattr = _testcapi.object_hasattrwitherror
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
self.assertTrue(xhasattr(obj, 'a'))
|
|
self.assertFalse(xhasattr(obj, 'b'))
|
|
self.assertTrue(xhasattr(obj, '\U0001f40d'))
|
|
|
|
self.assertRaises(RuntimeError, xhasattr, obj, 'evil')
|
|
self.assertRaises(TypeError, xhasattr, obj, 1)
|
|
# CRASHES xhasattr(obj, NULL)
|
|
# CRASHES xhasattr(NULL, 'a')
|
|
|
|
def test_object_hasattrstringwitherror(self):
|
|
hasattrstring = _testcapi.object_hasattrstringwitherror
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
self.assertTrue(hasattrstring(obj, b'a'))
|
|
self.assertFalse(hasattrstring(obj, b'b'))
|
|
self.assertTrue(hasattrstring(obj, '\U0001f40d'.encode()))
|
|
|
|
self.assertRaises(RuntimeError, hasattrstring, obj, b'evil')
|
|
self.assertRaises(UnicodeDecodeError, hasattrstring, obj, b'\xff')
|
|
# CRASHES hasattrstring(obj, NULL)
|
|
# CRASHES hasattrstring(NULL, b'a')
|
|
|
|
def test_object_setattr(self):
|
|
xsetattr = _testcapi.object_setattr
|
|
obj = TestObject()
|
|
xsetattr(obj, 'a', 5)
|
|
self.assertEqual(obj.a, 5)
|
|
xsetattr(obj, '\U0001f40d', 8)
|
|
self.assertEqual(getattr(obj, '\U0001f40d'), 8)
|
|
|
|
# PyObject_SetAttr(obj, attr_name, NULL) removes the attribute
|
|
xsetattr(obj, 'a', NULL)
|
|
self.assertFalse(hasattr(obj, 'a'))
|
|
self.assertRaises(AttributeError, xsetattr, obj, 'b', NULL)
|
|
self.assertRaises(RuntimeError, xsetattr, obj, 'evil', NULL)
|
|
|
|
self.assertRaises(RuntimeError, xsetattr, obj, 'evil', 'good')
|
|
self.assertRaises(AttributeError, xsetattr, 42, 'a', 5)
|
|
self.assertRaises(TypeError, xsetattr, obj, 1, 5)
|
|
# CRASHES xsetattr(obj, NULL, 5)
|
|
# CRASHES xsetattr(NULL, 'a', 5)
|
|
|
|
def test_object_setattrstring(self):
|
|
setattrstring = _testcapi.object_setattrstring
|
|
obj = TestObject()
|
|
setattrstring(obj, b'a', 5)
|
|
self.assertEqual(obj.a, 5)
|
|
setattrstring(obj, '\U0001f40d'.encode(), 8)
|
|
self.assertEqual(getattr(obj, '\U0001f40d'), 8)
|
|
|
|
# PyObject_SetAttrString(obj, attr_name, NULL) removes the attribute
|
|
setattrstring(obj, b'a', NULL)
|
|
self.assertFalse(hasattr(obj, 'a'))
|
|
self.assertRaises(AttributeError, setattrstring, obj, b'b', NULL)
|
|
self.assertRaises(RuntimeError, setattrstring, obj, b'evil', NULL)
|
|
|
|
self.assertRaises(RuntimeError, setattrstring, obj, b'evil', 'good')
|
|
self.assertRaises(AttributeError, setattrstring, 42, b'a', 5)
|
|
self.assertRaises(TypeError, setattrstring, obj, 1, 5)
|
|
self.assertRaises(UnicodeDecodeError, setattrstring, obj, b'\xff', 5)
|
|
# CRASHES setattrstring(obj, NULL, 5)
|
|
# CRASHES setattrstring(NULL, b'a', 5)
|
|
|
|
def test_object_delattr(self):
|
|
xdelattr = _testcapi.object_delattr
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
xdelattr(obj, 'a')
|
|
self.assertFalse(hasattr(obj, 'a'))
|
|
self.assertRaises(AttributeError, xdelattr, obj, 'b')
|
|
xdelattr(obj, '\U0001f40d')
|
|
self.assertFalse(hasattr(obj, '\U0001f40d'))
|
|
|
|
self.assertRaises(AttributeError, xdelattr, 42, 'numerator')
|
|
self.assertRaises(RuntimeError, xdelattr, obj, 'evil')
|
|
self.assertRaises(TypeError, xdelattr, obj, 1)
|
|
# CRASHES xdelattr(obj, NULL)
|
|
# CRASHES xdelattr(NULL, 'a')
|
|
|
|
def test_object_delattrstring(self):
|
|
delattrstring = _testcapi.object_delattrstring
|
|
obj = TestObject()
|
|
obj.a = 1
|
|
setattr(obj, '\U0001f40d', 2)
|
|
delattrstring(obj, b'a')
|
|
self.assertFalse(hasattr(obj, 'a'))
|
|
self.assertRaises(AttributeError, delattrstring, obj, b'b')
|
|
delattrstring(obj, '\U0001f40d'.encode())
|
|
self.assertFalse(hasattr(obj, '\U0001f40d'))
|
|
|
|
self.assertRaises(AttributeError, delattrstring, 42, b'numerator')
|
|
self.assertRaises(RuntimeError, delattrstring, obj, b'evil')
|
|
self.assertRaises(UnicodeDecodeError, delattrstring, obj, b'\xff')
|
|
# CRASHES delattrstring(obj, NULL)
|
|
# CRASHES delattrstring(NULL, b'a')
|
|
|
|
|
|
def test_mapping_check(self):
|
|
check = _testcapi.mapping_check
|
|
self.assertTrue(check({1: 2}))
|
|
self.assertTrue(check([1, 2]))
|
|
self.assertTrue(check((1, 2)))
|
|
self.assertTrue(check('abc'))
|
|
self.assertTrue(check(b'abc'))
|
|
self.assertFalse(check(42))
|
|
self.assertFalse(check(object()))
|
|
self.assertFalse(check(NULL))
|
|
|
|
def test_mapping_size(self):
|
|
for size in _testcapi.mapping_size, _testcapi.mapping_length:
|
|
self.assertEqual(size({1: 2}), 1)
|
|
self.assertEqual(size([1, 2]), 2)
|
|
self.assertEqual(size((1, 2)), 2)
|
|
self.assertEqual(size('abc'), 3)
|
|
self.assertEqual(size(b'abc'), 3)
|
|
|
|
self.assertRaises(TypeError, size, 42)
|
|
self.assertRaises(TypeError, size, object())
|
|
self.assertRaises(SystemError, size, NULL)
|
|
|
|
def test_object_getitem(self):
|
|
getitem = _testcapi.object_getitem
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertEqual(getitem(dct, 'a'), 1)
|
|
self.assertRaises(KeyError, getitem, dct, 'b')
|
|
self.assertEqual(getitem(dct, '\U0001f40d'), 2)
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertEqual(getitem(dct2, 'a'), 1)
|
|
self.assertRaises(KeyError, getitem, dct2, 'b')
|
|
|
|
self.assertEqual(getitem(['a', 'b', 'c'], 1), 'b')
|
|
|
|
self.assertRaises(TypeError, getitem, 42, 'a')
|
|
self.assertRaises(TypeError, getitem, {}, []) # unhashable
|
|
self.assertRaises(SystemError, getitem, {}, NULL)
|
|
self.assertRaises(IndexError, getitem, [], 1)
|
|
self.assertRaises(TypeError, getitem, [], 'a')
|
|
self.assertRaises(SystemError, getitem, NULL, 'a')
|
|
|
|
def test_mapping_getitemstring(self):
|
|
getitemstring = _testcapi.mapping_getitemstring
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertEqual(getitemstring(dct, b'a'), 1)
|
|
self.assertRaises(KeyError, getitemstring, dct, b'b')
|
|
self.assertEqual(getitemstring(dct, '\U0001f40d'.encode()), 2)
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertEqual(getitemstring(dct2, b'a'), 1)
|
|
self.assertRaises(KeyError, getitemstring, dct2, b'b')
|
|
|
|
self.assertRaises(TypeError, getitemstring, 42, b'a')
|
|
self.assertRaises(UnicodeDecodeError, getitemstring, {}, b'\xff')
|
|
self.assertRaises(SystemError, getitemstring, {}, NULL)
|
|
self.assertRaises(TypeError, getitemstring, [], b'a')
|
|
self.assertRaises(SystemError, getitemstring, NULL, b'a')
|
|
|
|
def test_mapping_getoptionalitem(self):
|
|
getitem = _testcapi.mapping_getoptionalitem
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertEqual(getitem(dct, 'a'), 1)
|
|
self.assertEqual(getitem(dct, 'b'), KeyError)
|
|
self.assertEqual(getitem(dct, '\U0001f40d'), 2)
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertEqual(getitem(dct2, 'a'), 1)
|
|
self.assertEqual(getitem(dct2, 'b'), KeyError)
|
|
|
|
self.assertEqual(getitem(['a', 'b', 'c'], 1), 'b')
|
|
|
|
self.assertRaises(TypeError, getitem, 42, 'a')
|
|
self.assertRaises(TypeError, getitem, {}, []) # unhashable
|
|
self.assertRaises(IndexError, getitem, [], 1)
|
|
self.assertRaises(TypeError, getitem, [], 'a')
|
|
# CRASHES getitem({}, NULL)
|
|
# CRASHES getitem(NULL, 'a')
|
|
|
|
def test_mapping_getoptionalitemstring(self):
|
|
getitemstring = _testcapi.mapping_getoptionalitemstring
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertEqual(getitemstring(dct, b'a'), 1)
|
|
self.assertEqual(getitemstring(dct, b'b'), KeyError)
|
|
self.assertEqual(getitemstring(dct, '\U0001f40d'.encode()), 2)
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertEqual(getitemstring(dct2, b'a'), 1)
|
|
self.assertEqual(getitemstring(dct2, b'b'), KeyError)
|
|
|
|
self.assertRaises(TypeError, getitemstring, 42, b'a')
|
|
self.assertRaises(UnicodeDecodeError, getitemstring, {}, b'\xff')
|
|
self.assertRaises(SystemError, getitemstring, {}, NULL)
|
|
self.assertRaises(TypeError, getitemstring, [], b'a')
|
|
# CRASHES getitemstring(NULL, b'a')
|
|
|
|
def test_mapping_haskey(self):
|
|
haskey = _testcapi.mapping_haskey
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertTrue(haskey(dct, 'a'))
|
|
self.assertFalse(haskey(dct, 'b'))
|
|
self.assertTrue(haskey(dct, '\U0001f40d'))
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertTrue(haskey(dct2, 'a'))
|
|
self.assertFalse(haskey(dct2, 'b'))
|
|
|
|
self.assertTrue(haskey(['a', 'b', 'c'], 1))
|
|
|
|
self.assertFalse(haskey(42, 'a'))
|
|
self.assertFalse(haskey({}, [])) # unhashable
|
|
self.assertFalse(haskey({}, NULL))
|
|
self.assertFalse(haskey([], 1))
|
|
self.assertFalse(haskey([], 'a'))
|
|
self.assertFalse(haskey(NULL, 'a'))
|
|
|
|
def test_mapping_haskeystring(self):
|
|
haskeystring = _testcapi.mapping_haskeystring
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertTrue(haskeystring(dct, b'a'))
|
|
self.assertFalse(haskeystring(dct, b'b'))
|
|
self.assertTrue(haskeystring(dct, '\U0001f40d'.encode()))
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertTrue(haskeystring(dct2, b'a'))
|
|
self.assertFalse(haskeystring(dct2, b'b'))
|
|
|
|
self.assertFalse(haskeystring(42, b'a'))
|
|
self.assertFalse(haskeystring({}, b'\xff'))
|
|
self.assertFalse(haskeystring({}, NULL))
|
|
self.assertFalse(haskeystring([], b'a'))
|
|
self.assertFalse(haskeystring(NULL, b'a'))
|
|
|
|
def test_mapping_haskeywitherror(self):
|
|
haskey = _testcapi.mapping_haskeywitherror
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertTrue(haskey(dct, 'a'))
|
|
self.assertFalse(haskey(dct, 'b'))
|
|
self.assertTrue(haskey(dct, '\U0001f40d'))
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertTrue(haskey(dct2, 'a'))
|
|
self.assertFalse(haskey(dct2, 'b'))
|
|
|
|
self.assertTrue(haskey(['a', 'b', 'c'], 1))
|
|
|
|
self.assertRaises(TypeError, haskey, 42, 'a')
|
|
self.assertRaises(TypeError, haskey, {}, []) # unhashable
|
|
self.assertRaises(IndexError, haskey, [], 1)
|
|
self.assertRaises(TypeError, haskey, [], 'a')
|
|
|
|
# CRASHES haskey({}, NULL))
|
|
# CRASHES haskey(NULL, 'a'))
|
|
|
|
def test_mapping_haskeystringwitherror(self):
|
|
haskeystring = _testcapi.mapping_haskeystringwitherror
|
|
dct = {'a': 1, '\U0001f40d': 2}
|
|
self.assertTrue(haskeystring(dct, b'a'))
|
|
self.assertFalse(haskeystring(dct, b'b'))
|
|
self.assertTrue(haskeystring(dct, '\U0001f40d'.encode()))
|
|
|
|
dct2 = ProxyGetItem(dct)
|
|
self.assertTrue(haskeystring(dct2, b'a'))
|
|
self.assertFalse(haskeystring(dct2, b'b'))
|
|
|
|
self.assertRaises(TypeError, haskeystring, 42, b'a')
|
|
self.assertRaises(UnicodeDecodeError, haskeystring, {}, b'\xff')
|
|
self.assertRaises(SystemError, haskeystring, {}, NULL)
|
|
self.assertRaises(TypeError, haskeystring, [], b'a')
|
|
# CRASHES haskeystring(NULL, b'a')
|
|
|
|
def test_object_setitem(self):
|
|
setitem = _testcapi.object_setitem
|
|
dct = {}
|
|
setitem(dct, 'a', 5)
|
|
self.assertEqual(dct, {'a': 5})
|
|
setitem(dct, '\U0001f40d', 8)
|
|
self.assertEqual(dct, {'a': 5, '\U0001f40d': 8})
|
|
|
|
dct = {}
|
|
dct2 = ProxySetItem(dct)
|
|
setitem(dct2, 'a', 5)
|
|
self.assertEqual(dct, {'a': 5})
|
|
|
|
lst = ['a', 'b', 'c']
|
|
setitem(lst, 1, 'x')
|
|
self.assertEqual(lst, ['a', 'x', 'c'])
|
|
|
|
self.assertRaises(TypeError, setitem, 42, 'a', 5)
|
|
self.assertRaises(TypeError, setitem, {}, [], 5) # unhashable
|
|
self.assertRaises(SystemError, setitem, {}, NULL, 5)
|
|
self.assertRaises(SystemError, setitem, {}, 'a', NULL)
|
|
self.assertRaises(IndexError, setitem, [], 1, 5)
|
|
self.assertRaises(TypeError, setitem, [], 'a', 5)
|
|
self.assertRaises(TypeError, setitem, (), 1, 5)
|
|
self.assertRaises(SystemError, setitem, NULL, 'a', 5)
|
|
|
|
def test_mapping_setitemstring(self):
|
|
setitemstring = _testcapi.mapping_setitemstring
|
|
dct = {}
|
|
setitemstring(dct, b'a', 5)
|
|
self.assertEqual(dct, {'a': 5})
|
|
setitemstring(dct, '\U0001f40d'.encode(), 8)
|
|
self.assertEqual(dct, {'a': 5, '\U0001f40d': 8})
|
|
|
|
dct = {}
|
|
dct2 = ProxySetItem(dct)
|
|
setitemstring(dct2, b'a', 5)
|
|
self.assertEqual(dct, {'a': 5})
|
|
|
|
self.assertRaises(TypeError, setitemstring, 42, b'a', 5)
|
|
self.assertRaises(UnicodeDecodeError, setitemstring, {}, b'\xff', 5)
|
|
self.assertRaises(SystemError, setitemstring, {}, NULL, 5)
|
|
self.assertRaises(SystemError, setitemstring, {}, b'a', NULL)
|
|
self.assertRaises(TypeError, setitemstring, [], b'a', 5)
|
|
self.assertRaises(SystemError, setitemstring, NULL, b'a', 5)
|
|
|
|
def test_object_delitem(self):
|
|
for delitem in _testcapi.object_delitem, _testcapi.mapping_delitem:
|
|
dct = {'a': 1, 'c': 2, '\U0001f40d': 3}
|
|
delitem(dct, 'a')
|
|
self.assertEqual(dct, {'c': 2, '\U0001f40d': 3})
|
|
self.assertRaises(KeyError, delitem, dct, 'b')
|
|
delitem(dct, '\U0001f40d')
|
|
self.assertEqual(dct, {'c': 2})
|
|
|
|
dct = {'a': 1, 'c': 2}
|
|
dct2 = ProxyDelItem(dct)
|
|
delitem(dct2, 'a')
|
|
self.assertEqual(dct, {'c': 2})
|
|
self.assertRaises(KeyError, delitem, dct2, 'b')
|
|
|
|
lst = ['a', 'b', 'c']
|
|
delitem(lst, 1)
|
|
self.assertEqual(lst, ['a', 'c'])
|
|
|
|
self.assertRaises(TypeError, delitem, 42, 'a')
|
|
self.assertRaises(TypeError, delitem, {}, []) # unhashable
|
|
self.assertRaises(SystemError, delitem, {}, NULL)
|
|
self.assertRaises(IndexError, delitem, [], 1)
|
|
self.assertRaises(TypeError, delitem, [], 'a')
|
|
self.assertRaises(SystemError, delitem, NULL, 'a')
|
|
|
|
def test_mapping_delitemstring(self):
|
|
delitemstring = _testcapi.mapping_delitemstring
|
|
dct = {'a': 1, 'c': 2, '\U0001f40d': 3}
|
|
delitemstring(dct, b'a')
|
|
self.assertEqual(dct, {'c': 2, '\U0001f40d': 3})
|
|
self.assertRaises(KeyError, delitemstring, dct, b'b')
|
|
delitemstring(dct, '\U0001f40d'.encode())
|
|
self.assertEqual(dct, {'c': 2})
|
|
|
|
dct = {'a': 1, 'c': 2}
|
|
dct2 = ProxyDelItem(dct)
|
|
delitemstring(dct2, b'a')
|
|
self.assertEqual(dct, {'c': 2})
|
|
self.assertRaises(KeyError, delitemstring, dct2, b'b')
|
|
|
|
self.assertRaises(TypeError, delitemstring, 42, b'a')
|
|
self.assertRaises(UnicodeDecodeError, delitemstring, {}, b'\xff')
|
|
self.assertRaises(SystemError, delitemstring, {}, NULL)
|
|
self.assertRaises(TypeError, delitemstring, [], b'a')
|
|
self.assertRaises(SystemError, delitemstring, NULL, b'a')
|
|
|
|
def test_mapping_keys_valuesitems(self):
|
|
class Mapping1(dict):
|
|
def keys(self):
|
|
return list(super().keys())
|
|
def values(self):
|
|
return list(super().values())
|
|
def items(self):
|
|
return list(super().items())
|
|
class Mapping2(dict):
|
|
def keys(self):
|
|
return tuple(super().keys())
|
|
def values(self):
|
|
return tuple(super().values())
|
|
def items(self):
|
|
return tuple(super().items())
|
|
dict_obj = {'foo': 1, 'bar': 2, 'spam': 3}
|
|
|
|
for mapping in [{}, OrderedDict(), Mapping1(), Mapping2(),
|
|
dict_obj, OrderedDict(dict_obj),
|
|
Mapping1(dict_obj), Mapping2(dict_obj)]:
|
|
self.assertListEqual(_testcapi.mapping_keys(mapping),
|
|
list(mapping.keys()))
|
|
self.assertListEqual(_testcapi.mapping_values(mapping),
|
|
list(mapping.values()))
|
|
self.assertListEqual(_testcapi.mapping_items(mapping),
|
|
list(mapping.items()))
|
|
|
|
def test_mapping_keys_valuesitems_bad_arg(self):
|
|
self.assertRaises(AttributeError, _testcapi.mapping_keys, object())
|
|
self.assertRaises(AttributeError, _testcapi.mapping_values, object())
|
|
self.assertRaises(AttributeError, _testcapi.mapping_items, object())
|
|
self.assertRaises(AttributeError, _testcapi.mapping_keys, [])
|
|
self.assertRaises(AttributeError, _testcapi.mapping_values, [])
|
|
self.assertRaises(AttributeError, _testcapi.mapping_items, [])
|
|
self.assertRaises(SystemError, _testcapi.mapping_keys, NULL)
|
|
self.assertRaises(SystemError, _testcapi.mapping_values, NULL)
|
|
self.assertRaises(SystemError, _testcapi.mapping_items, NULL)
|
|
|
|
class BadMapping:
|
|
def keys(self):
|
|
return None
|
|
def values(self):
|
|
return None
|
|
def items(self):
|
|
return None
|
|
bad_mapping = BadMapping()
|
|
self.assertRaises(TypeError, _testcapi.mapping_keys, bad_mapping)
|
|
self.assertRaises(TypeError, _testcapi.mapping_values, bad_mapping)
|
|
self.assertRaises(TypeError, _testcapi.mapping_items, bad_mapping)
|
|
|
|
def test_sequence_check(self):
|
|
check = _testcapi.sequence_check
|
|
self.assertFalse(check({1: 2}))
|
|
self.assertTrue(check([1, 2]))
|
|
self.assertTrue(check((1, 2)))
|
|
self.assertTrue(check('abc'))
|
|
self.assertTrue(check(b'abc'))
|
|
self.assertFalse(check(42))
|
|
self.assertFalse(check(object()))
|
|
# CRASHES check(NULL)
|
|
|
|
def test_sequence_size(self):
|
|
for size in _testcapi.sequence_size, _testcapi.sequence_length:
|
|
self.assertEqual(size([1, 2]), 2)
|
|
self.assertEqual(size((1, 2)), 2)
|
|
self.assertEqual(size('abc'), 3)
|
|
self.assertEqual(size(b'abc'), 3)
|
|
|
|
self.assertRaises(TypeError, size, {})
|
|
self.assertRaises(TypeError, size, 42)
|
|
self.assertRaises(TypeError, size, object())
|
|
self.assertRaises(SystemError, size, NULL)
|
|
|
|
def test_sequence_getitem(self):
|
|
getitem = _testcapi.sequence_getitem
|
|
lst = ['a', 'b', 'c']
|
|
self.assertEqual(getitem(lst, 1), 'b')
|
|
self.assertEqual(getitem(lst, -1), 'c')
|
|
self.assertRaises(IndexError, getitem, lst, 3)
|
|
|
|
self.assertRaises(TypeError, getitem, 42, 1)
|
|
self.assertRaises(TypeError, getitem, {}, 1)
|
|
self.assertRaises(SystemError, getitem, NULL, 1)
|
|
|
|
def test_sequence_concat(self):
|
|
concat = _testcapi.sequence_concat
|
|
self.assertEqual(concat(['a', 'b'], [1, 2]), ['a', 'b', 1, 2])
|
|
self.assertEqual(concat(('a', 'b'), (1, 2)), ('a', 'b', 1, 2))
|
|
|
|
self.assertRaises(TypeError, concat, [], ())
|
|
self.assertRaises(TypeError, concat, (), [])
|
|
self.assertRaises(TypeError, concat, [], 42)
|
|
self.assertRaises(TypeError, concat, 42, [])
|
|
self.assertRaises(TypeError, concat, 42, 43)
|
|
self.assertRaises(SystemError, concat, [], NULL)
|
|
self.assertRaises(SystemError, concat, NULL, [])
|
|
|
|
def test_sequence_repeat(self):
|
|
repeat = _testcapi.sequence_repeat
|
|
self.assertEqual(repeat(['a', 'b'], 2), ['a', 'b', 'a', 'b'])
|
|
self.assertEqual(repeat(('a', 'b'), 2), ('a', 'b', 'a', 'b'))
|
|
self.assertEqual(repeat(['a', 'b'], 0), [])
|
|
self.assertEqual(repeat(['a', 'b'], -1), [])
|
|
|
|
self.assertRaises(TypeError, repeat, set(), 2)
|
|
self.assertRaises(TypeError, repeat, 42, 2)
|
|
self.assertRaises(SystemError, repeat, NULL, 2)
|
|
|
|
def test_sequence_inplaceconcat(self):
|
|
inplaceconcat = _testcapi.sequence_inplaceconcat
|
|
lst = ['a', 'b']
|
|
res = inplaceconcat(lst, [1, 2])
|
|
self.assertEqual(res, ['a', 'b', 1, 2])
|
|
self.assertIs(res, lst)
|
|
lst = ['a', 'b']
|
|
res = inplaceconcat(lst, (1, 2))
|
|
self.assertEqual(res, ['a', 'b', 1, 2])
|
|
self.assertIs(res, lst)
|
|
self.assertEqual(inplaceconcat(('a', 'b'), (1, 2)), ('a', 'b', 1, 2))
|
|
|
|
self.assertRaises(TypeError, inplaceconcat, (), [])
|
|
self.assertRaises(TypeError, inplaceconcat, [], 42)
|
|
self.assertRaises(TypeError, inplaceconcat, 42, [])
|
|
self.assertRaises(TypeError, inplaceconcat, 42, 43)
|
|
self.assertRaises(SystemError, inplaceconcat, [], NULL)
|
|
self.assertRaises(SystemError, inplaceconcat, NULL, [])
|
|
|
|
def test_sequence_inplacerepeat(self):
|
|
inplacerepeat = _testcapi.sequence_inplacerepeat
|
|
lst = ['a', 'b']
|
|
res = inplacerepeat(lst, 2)
|
|
self.assertEqual(res, ['a', 'b', 'a', 'b'])
|
|
self.assertIs(res, lst)
|
|
self.assertEqual(inplacerepeat(('a', 'b'), 2), ('a', 'b', 'a', 'b'))
|
|
self.assertEqual(inplacerepeat(['a', 'b'], 0), [])
|
|
self.assertEqual(inplacerepeat(['a', 'b'], -1), [])
|
|
|
|
self.assertRaises(TypeError, inplacerepeat, set(), 2)
|
|
self.assertRaises(TypeError, inplacerepeat, 42, 2)
|
|
self.assertRaises(SystemError, inplacerepeat, NULL, 2)
|
|
|
|
def test_sequence_setitem(self):
|
|
setitem = _testcapi.sequence_setitem
|
|
lst = ['a', 'b', 'c']
|
|
setitem(lst, 1, 'x')
|
|
self.assertEqual(lst, ['a', 'x', 'c'])
|
|
setitem(lst, -1, 'y')
|
|
self.assertEqual(lst, ['a', 'x', 'y'])
|
|
|
|
setitem(lst, 0, NULL)
|
|
self.assertEqual(lst, ['x', 'y'])
|
|
self.assertRaises(IndexError, setitem, lst, 3, 'x')
|
|
|
|
self.assertRaises(TypeError, setitem, 42, 1, 'x')
|
|
self.assertRaises(TypeError, setitem, {}, 1, 'x')
|
|
self.assertRaises(SystemError, setitem, NULL, 1, 'x')
|
|
|
|
def test_sequence_delitem(self):
|
|
delitem = _testcapi.sequence_delitem
|
|
lst = ['a', 'b', 'c']
|
|
delitem(lst, 1)
|
|
self.assertEqual(lst, ['a', 'c'])
|
|
delitem(lst, -1)
|
|
self.assertEqual(lst, ['a'])
|
|
self.assertRaises(IndexError, delitem, lst, 3)
|
|
|
|
self.assertRaises(TypeError, delitem, 42, 1)
|
|
self.assertRaises(TypeError, delitem, {}, 1)
|
|
self.assertRaises(SystemError, delitem, NULL, 1)
|
|
|
|
def test_sequence_setslice(self):
|
|
setslice = _testcapi.sequence_setslice
|
|
|
|
# Correct case:
|
|
data = [1, 2, 3, 4, 5]
|
|
data_copy = data.copy()
|
|
|
|
setslice(data, 1, 3, [8, 9])
|
|
data_copy[1:3] = [8, 9]
|
|
self.assertEqual(data, data_copy)
|
|
self.assertEqual(data, [1, 8, 9, 4, 5])
|
|
|
|
# Custom class:
|
|
class Custom:
|
|
def __setitem__(self, index, value):
|
|
self.index = index
|
|
self.value = value
|
|
|
|
c = Custom()
|
|
setslice(c, 0, 5, 'abc')
|
|
self.assertEqual(c.index, slice(0, 5))
|
|
self.assertEqual(c.value, 'abc')
|
|
|
|
# Immutable sequences must raise:
|
|
bad_seq1 = (1, 2, 3, 4)
|
|
self.assertRaises(TypeError, setslice, bad_seq1, 1, 3, (8, 9))
|
|
self.assertEqual(bad_seq1, (1, 2, 3, 4))
|
|
|
|
bad_seq2 = 'abcd'
|
|
self.assertRaises(TypeError, setslice, bad_seq2, 1, 3, 'xy')
|
|
self.assertEqual(bad_seq2, 'abcd')
|
|
|
|
# Not a sequence:
|
|
self.assertRaises(TypeError, setslice, object(), 1, 3, 'xy')
|
|
self.assertRaises(SystemError, setslice, NULL, 1, 3, 'xy')
|
|
|
|
data_copy = data.copy()
|
|
setslice(data_copy, 1, 3, NULL)
|
|
self.assertEqual(data_copy, [1, 4, 5])
|
|
|
|
def test_sequence_delslice(self):
|
|
delslice = _testcapi.sequence_delslice
|
|
|
|
# Correct case:
|
|
data = [1, 2, 3, 4, 5]
|
|
data_copy = data.copy()
|
|
|
|
delslice(data, 1, 3)
|
|
del data_copy[1:3]
|
|
self.assertEqual(data, data_copy)
|
|
self.assertEqual(data, [1, 4, 5])
|
|
|
|
# Custom class:
|
|
class Custom:
|
|
def __delitem__(self, index):
|
|
self.index = index
|
|
|
|
c = Custom()
|
|
delslice(c, 0, 5)
|
|
self.assertEqual(c.index, slice(0, 5))
|
|
|
|
# Immutable sequences must raise:
|
|
bad_seq1 = (1, 2, 3, 4)
|
|
self.assertRaises(TypeError, delslice, bad_seq1, 1, 3)
|
|
self.assertEqual(bad_seq1, (1, 2, 3, 4))
|
|
|
|
bad_seq2 = 'abcd'
|
|
self.assertRaises(TypeError, delslice, bad_seq2, 1, 3)
|
|
self.assertEqual(bad_seq2, 'abcd')
|
|
|
|
# Not a sequence:
|
|
self.assertRaises(TypeError, delslice, object(), 1, 3)
|
|
self.assertRaises(SystemError, delslice, NULL, 1, 3)
|
|
|
|
mapping = {1: 'a', 2: 'b', 3: 'c'}
|
|
self.assertRaises(KeyError, delslice, mapping, 1, 3)
|
|
self.assertEqual(mapping, {1: 'a', 2: 'b', 3: 'c'})
|
|
|
|
def test_sequence_count(self):
|
|
count = _testcapi.sequence_count
|
|
|
|
lst = ['a', 'b', 'a']
|
|
self.assertEqual(count(lst, 'a'), 2)
|
|
self.assertEqual(count(lst, 'c'), 0)
|
|
self.assertEqual(count(iter(lst), 'a'), 2)
|
|
self.assertEqual(count(iter(lst), 'c'), 0)
|
|
self.assertEqual(count({'a': 2}, 'a'), 1)
|
|
|
|
self.assertRaises(TypeError, count, 42, 'a')
|
|
self.assertRaises(SystemError, count, [], NULL)
|
|
self.assertRaises(SystemError, count, [1], NULL)
|
|
self.assertRaises(SystemError, count, NULL, 'a')
|
|
|
|
def test_sequence_contains(self):
|
|
contains = _testcapi.sequence_contains
|
|
|
|
lst = ['a', 'b', 'a']
|
|
self.assertEqual(contains(lst, 'a'), 1)
|
|
self.assertEqual(contains(lst, 'c'), 0)
|
|
self.assertEqual(contains(iter(lst), 'a'), 1)
|
|
self.assertEqual(contains(iter(lst), 'c'), 0)
|
|
self.assertEqual(contains({'a': 2}, 'a'), 1)
|
|
|
|
# XXX Only for empty sequences. Should be SystemError?
|
|
self.assertEqual(contains([], NULL), 0)
|
|
|
|
self.assertRaises(TypeError, contains, 42, 'a')
|
|
self.assertRaises(SystemError, contains, [1], NULL)
|
|
# CRASHES contains({}, NULL)
|
|
# CRASHES contains(set(), NULL)
|
|
# CRASHES contains(NULL, 'a')
|
|
|
|
def test_sequence_index(self):
|
|
index = _testcapi.sequence_index
|
|
|
|
lst = ['a', 'b', 'a']
|
|
self.assertEqual(index(lst, 'a'), 0)
|
|
self.assertEqual(index(lst, 'b'), 1)
|
|
self.assertRaises(ValueError, index, lst, 'c')
|
|
self.assertEqual(index(iter(lst), 'a'), 0)
|
|
self.assertEqual(index(iter(lst), 'b'), 1)
|
|
self.assertRaises(ValueError, index, iter(lst), 'c')
|
|
dct = {'a': 2, 'b': 3}
|
|
self.assertEqual(index(dct, 'a'), 0)
|
|
self.assertEqual(index(dct, 'b'), 1)
|
|
self.assertRaises(ValueError, index, dct, 'c')
|
|
|
|
self.assertRaises(TypeError, index, 42, 'a')
|
|
self.assertRaises(SystemError, index, [], NULL)
|
|
self.assertRaises(SystemError, index, [1], NULL)
|
|
self.assertRaises(SystemError, index, NULL, 'a')
|
|
|
|
def test_sequence_list(self):
|
|
xlist = _testcapi.sequence_list
|
|
self.assertEqual(xlist(['a', 'b', 'c']), ['a', 'b', 'c'])
|
|
self.assertEqual(xlist(('a', 'b', 'c')), ['a', 'b', 'c'])
|
|
self.assertEqual(xlist(iter(['a', 'b', 'c'])), ['a', 'b', 'c'])
|
|
self.assertEqual(xlist(gen()), ['a', 'b', 'c'])
|
|
|
|
self.assertRaises(TypeError, xlist, 42)
|
|
self.assertRaises(SystemError, xlist, NULL)
|
|
|
|
def test_sequence_tuple(self):
|
|
xtuple = _testcapi.sequence_tuple
|
|
self.assertEqual(xtuple(['a', 'b', 'c']), ('a', 'b', 'c'))
|
|
self.assertEqual(xtuple(('a', 'b', 'c')), ('a', 'b', 'c'))
|
|
self.assertEqual(xtuple(iter(['a', 'b', 'c'])), ('a', 'b', 'c'))
|
|
self.assertEqual(xtuple(gen()), ('a', 'b', 'c'))
|
|
|
|
self.assertRaises(TypeError, xtuple, 42)
|
|
self.assertRaises(SystemError, xtuple, NULL)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|