10c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi# tests common to dict and UserDict 20c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiimport unittest 30c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiimport UserDict 40c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiimport test_support 50c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 60c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 70c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiclass BasicTestMappingProtocol(unittest.TestCase): 80c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # This base class can be used to check that an object conforms to the 90c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # mapping protocol 100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Functions that can be useful to override to adapt to dictionary 120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # semantics 130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi type2test = None # which class is being tested (overwrite in subclasses) 140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def _reference(self): 160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi """Return a dictionary of values which are invariant by storage 170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi in the object under test.""" 180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return {1:2, "key1":"value1", "key2":(1,2,3)} 190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def _empty_mapping(self): 200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi """Return an empty mapping object""" 210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self.type2test() 220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def _full_mapping(self, data): 230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi """Return a mapping object with the value contained in data 240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi dictionary""" 250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = self._empty_mapping() 260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for key, value in data.items(): 270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x[key] = value 280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return x 290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self, *args, **kw): 310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi unittest.TestCase.__init__(self, *args, **kw) 320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.reference = self._reference().copy() 330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # A (key, value) pair not in the mapping 350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi key, value = self.reference.popitem() 360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.other = {key:value} 370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # A (key, value) pair in the mapping 390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi key, value = self.reference.popitem() 400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.inmapping = {key:value} 410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.reference[key] = value 420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_read(self): 440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Test for read only operations on mapping 450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p = self._empty_mapping() 460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p1 = dict(p) #workaround for singleton objects 470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping(self.reference) 480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if d is p: 490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p = p1 500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #Indexing 510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for key, value in self.reference.items(): 520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d[key], value) 530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi knownkey = self.other.keys()[0] 540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, lambda:d[knownkey]) 550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #len 560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(p), 0) 570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(d), len(self.reference)) 580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #in 590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for k in self.reference: 600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn(k, d) 610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for k in self.other: 620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(k, d) 630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #has_key 640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi with test_support.check_py3k_warnings(quiet=True): 650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for k in self.reference: 660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.has_key(k)) 670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for k in self.other: 680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertFalse(d.has_key(k)) 690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #cmp 700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(cmp(p,p), 0) 710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(cmp(d,d), 0) 720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(cmp(p,d), -1) 730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(cmp(d,p), 1) 740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #__non__zero__ 750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if p: self.fail("Empty mapping must compare to False") 760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if not d: self.fail("Full mapping must compare to True") 770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # keys(), items(), iterkeys() ... 780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def check_iterandlist(iter, lst, ref): 790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(hasattr(iter, 'next')) 800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(hasattr(iter, '__iter__')) 810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = list(iter) 820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(set(x)==set(lst)==set(ref)) 830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi check_iterandlist(d.iterkeys(), d.keys(), self.reference.keys()) 840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi check_iterandlist(iter(d), d.keys(), self.reference.keys()) 850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi check_iterandlist(d.itervalues(), d.values(), self.reference.values()) 860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi check_iterandlist(d.iteritems(), d.items(), self.reference.items()) 870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #get 880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi key, value = d.iteritems().next() 890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi knownkey, knownvalue = self.other.iteritems().next() 900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(key, knownvalue), value) 910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(knownkey, knownvalue), knownvalue) 920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(knownkey, d) 930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_write(self): 950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Test for write operations on mapping 960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p = self._empty_mapping() 970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #Indexing 980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for key, value in self.reference.items(): 990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p[key] = value 1000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(p[key], value) 1010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for key in self.reference.keys(): 1020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi del p[key] 1030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, lambda:p[key]) 1040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p = self._empty_mapping() 1050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #update 1060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p.update(self.reference) 1070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(dict(p), self.reference) 1080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi items = p.items() 1090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p = self._empty_mapping() 1100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p.update(items) 1110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(dict(p), self.reference) 1120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping(self.reference) 1130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #setdefault 1140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi key, value = d.iteritems().next() 1150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi knownkey, knownvalue = self.other.iteritems().next() 1160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.setdefault(key, knownvalue), value) 1170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d[key], value) 1180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.setdefault(knownkey, knownvalue), knownvalue) 1190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d[knownkey], knownvalue) 1200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #pop 1210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(knownkey), knownvalue) 1220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(knownkey, d) 1230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, d.pop, knownkey) 1240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi default = 909 1250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[knownkey] = knownvalue 1260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(knownkey, default), knownvalue) 1270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(knownkey, d) 1280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(knownkey, default), default) 1290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi #popitem 1300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi key, value = d.popitem() 1310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(key, d) 1320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(value, self.reference[key]) 1330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi p=self._empty_mapping() 1340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, p.popitem) 1350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_constructor(self): 1370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(self._empty_mapping(), self._empty_mapping()) 1380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_bool(self): 1400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not self._empty_mapping()) 1410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(self.reference) 1420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(bool(self._empty_mapping()) is False) 1430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(bool(self.reference) is True) 1440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_keys(self): 1460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.keys(), []) 1480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self.reference 1490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn(self.inmapping.keys()[0], d.keys()) 1500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn(self.other.keys()[0], d.keys()) 1510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.keys, None) 1520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_values(self): 1540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.values(), []) 1560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.values, None) 1580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_items(self): 1600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.items(), []) 1620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.items, None) 1640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_len(self): 1660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(d), 0) 1680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_getitem(self): 1700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self.reference 1710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d[self.inmapping.keys()[0]], self.inmapping.values()[0]) 1720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.__getitem__) 1740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_update(self): 1760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # mapping argument 1770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(self.other) 1790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.items(), self.other.items()) 1800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # No argument 1820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update() 1840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, self._empty_mapping()) 1850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # item sequence 1870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(self.other.items()) 1890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.items(), self.other.items()) 1900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Iterator 1920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 1930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(self.other.iteritems()) 1940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.items(), self.other.items()) 1950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 1960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # FIXME: Doesn't work with UserDict 1970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # self.assertRaises((TypeError, AttributeError), d.update, None) 1980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises((TypeError, AttributeError), d.update, 42) 1990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi outerself = self 2010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class SimpleUserDict: 2020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self): 2030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.d = outerself.reference 2040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def keys(self): 2050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self.d.keys() 2060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __getitem__(self, i): 2070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self.d[i] 2080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.clear() 2090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(SimpleUserDict()) 2100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi i1 = d.items() 2110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi i2 = self.reference.items() 2120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def safe_sort_key(kv): 2140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k, v = kv 2150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return id(type(k)), id(type(v)), k, v 2160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi i1.sort(key=safe_sort_key) 2170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi i2.sort(key=safe_sort_key) 2180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(i1, i2) 2190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 2210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 2230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class FailingUserDict: 2240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def keys(self): 2250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc 2260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.update, FailingUserDict()) 2270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.clear() 2290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class FailingUserDict: 2310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def keys(self): 2320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BogonIter: 2330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self): 2340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.i = 1 2350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __iter__(self): 2360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self 2370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def next(self): 2380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if self.i: 2390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.i = 0 2400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 'a' 2410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc 2420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return BogonIter() 2430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __getitem__(self, key): 2440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return key 2450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.update, FailingUserDict()) 2460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class FailingUserDict: 2480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def keys(self): 2490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BogonIter: 2500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self): 2510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.i = ord('a') 2520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __iter__(self): 2530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self 2540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def next(self): 2550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if self.i <= ord('z'): 2560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi rtn = chr(self.i) 2570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.i += 1 2580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return rtn 2590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise StopIteration 2600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return BogonIter() 2610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __getitem__(self, key): 2620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc 2630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.update, FailingUserDict()) 2640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 2660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class badseq(object): 2670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __iter__(self): 2680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self 2690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def next(self): 2700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 2710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.update, badseq()) 2730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(ValueError, d.update, [(1, 2, 3)]) 2750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # no test_fromkeys or test_copy as both os.environ and selves don't support it 2770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_get(self): 2790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 2800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.get(self.other.keys()[0]) is None) 2810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(self.other.keys()[0], 3), 3) 2820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self.reference 2830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.get(self.other.keys()[0]) is None) 2840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(self.other.keys()[0], 3), 3) 2850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(self.inmapping.keys()[0]), self.inmapping.values()[0]) 2860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get(self.inmapping.keys()[0], 3), self.inmapping.values()[0]) 2870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.get) 2880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.get, None, None, None) 2890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_setdefault(self): 2910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 2920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.setdefault) 2930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_popitem(self): 2950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 2960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, d.popitem) 2970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.popitem, 42) 2980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 2990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_pop(self): 3000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 3010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k, v = self.inmapping.items()[0] 3020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[k] = v 3030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, d.pop, self.other.keys()[0]) 3040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(k), v) 3060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(d), 0) 3070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, d.pop, k) 3090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiclass TestMappingProtocol(BasicTestMappingProtocol): 3120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_constructor(self): 3130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_constructor(self) 3140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(self._empty_mapping() is not self._empty_mapping()) 3150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(self.type2test(x=1, y=2), {"x": 1, "y": 2}) 3160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_bool(self): 3180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_bool(self) 3190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not self._empty_mapping()) 3200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(self._full_mapping({"x": "y"})) 3210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(bool(self._empty_mapping()) is False) 3220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(bool(self._full_mapping({"x": "y"})) is True) 3230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_keys(self): 3250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_keys(self) 3260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 3270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.keys(), []) 3280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a': 1, 'b': 2}) 3290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k = d.keys() 3300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn('a', k) 3310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn('b', k) 3320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn('c', k) 3330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_values(self): 3350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_values(self) 3360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1:2}) 3370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.values(), [2]) 3380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_items(self): 3400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_items(self) 3410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1:2}) 3430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.items(), [(1, 2)]) 3440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_has_key(self): 3460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 3470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not d.has_key('a')) 3480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a': 1, 'b': 2}) 3490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k = d.keys() 3500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k.sort(key=lambda k: (id(type(k)), k)) 3510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(k, ['a', 'b']) 3520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.has_key) 3540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_contains(self): 3560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 3570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn('a', d) 3580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not ('a' in d)) 3590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue('a' not in d) 3600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a': 1, 'b': 2}) 3610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn('a', d) 3620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIn('b', d) 3630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertNotIn('c', d) 3640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.__contains__) 3660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_len(self): 3680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_len(self) 3690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a': 1, 'b': 2}) 3700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(d), 2) 3710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_getitem(self): 3730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_getitem(self) 3740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a': 1, 'b': 2}) 3750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d['a'], 1) 3760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d['b'], 2) 3770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d['c'] = 3 3780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d['a'] = 4 3790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d['c'], 3) 3800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d['a'], 4) 3810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi del d['b'] 3820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {'a': 4, 'c': 3}) 3830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.__getitem__) 3850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_clear(self): 3870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1:1, 2:2, 3:3}) 3880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.clear() 3890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {}) 3900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.clear, None) 3920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 3930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_update(self): 3940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_update(self) 3950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # mapping argument 3960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 3970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update({1:100}) 3980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update({2:20}) 3990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update({1:1, 2:2, 3:3}) 4000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {1:1, 2:2, 3:3}) 4010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # no argument 4030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update() 4040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {1:1, 2:2, 3:3}) 4050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # keyword arguments 4070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(x=100) 4090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(y=20) 4100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(x=1, y=2, z=3) 4110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {"x":1, "y":2, "z":3}) 4120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # item sequence 4140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update([("x", 100), ("y", 20)]) 4160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {"x":100, "y":20}) 4170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Both item sequence and keyword arguments 4190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update([("x", 100), ("y", 20)], x=1, y=2) 4210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {"x":1, "y":2}) 4220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # iterator 4240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1:3, 2:4}) 4250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(self._full_mapping({1:2, 3:4, 5:6}).iteritems()) 4260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {1:2, 2:4, 3:4, 5:6}) 4270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class SimpleUserDict: 4290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self): 4300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.d = {1:1, 2:2, 3:3} 4310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def keys(self): 4320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self.d.keys() 4330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __getitem__(self, i): 4340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self.d[i] 4350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.clear() 4360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.update(SimpleUserDict()) 4370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d, {1:1, 2:2, 3:3}) 4380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_fromkeys(self): 4400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(self.type2test.fromkeys('abc'), {'a':None, 'b':None, 'c':None}) 4410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not(d.fromkeys('abc') is d)) 4430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.fromkeys('abc'), {'a':None, 'b':None, 'c':None}) 4440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.fromkeys((4,5),0), {4:0, 5:0}) 4450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.fromkeys([]), {}) 4460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def g(): 4470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi yield 1 4480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.fromkeys(g()), {1:None}) 4490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, {}.fromkeys, 3) 4500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class dictlike(self.type2test): pass 4510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(dictlike.fromkeys('a'), {'a':None}) 4520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(dictlike().fromkeys('a'), {'a':None}) 4530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(dictlike.fromkeys('a').__class__ is dictlike) 4540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(dictlike().fromkeys('a').__class__ is dictlike) 4550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # FIXME: the following won't work with UserDict, because it's an old style class 4560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # self.assertTrue(type(dictlike.fromkeys('a')) is dictlike) 4570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class mydict(self.type2test): 4580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __new__(cls): 4590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return UserDict.UserDict() 4600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi ud = mydict.fromkeys('ab') 4610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(ud, {'a':None, 'b':None}) 4620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # FIXME: the following won't work with UserDict, because it's an old style class 4630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # self.assertIsInstance(ud, UserDict.UserDict) 4640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, dict.fromkeys) 4650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 4670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class baddict1(self.type2test): 4690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __init__(self): 4700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 4710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, baddict1.fromkeys, [1]) 4730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadSeq(object): 4750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __iter__(self): 4760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return self 4770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def next(self): 4780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 4790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, self.type2test.fromkeys, BadSeq()) 4810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class baddict2(self.type2test): 4830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __setitem__(self, key, value): 4840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 4850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, baddict2.fromkeys, [1]) 4870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_copy(self): 4890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1:1, 2:2, 3:3}) 4900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.copy(), {1:1, 2:2, 3:3}) 4910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.copy(), d) 4930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIsInstance(d.copy(), d.__class__) 4940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(TypeError, d.copy, None) 4950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 4960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_get(self): 4970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_get(self) 4980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 4990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.get('c') is None) 5000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get('c', 3), 3) 5010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({'a' : 1, 'b' : 2}) 5020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.get('c') is None) 5030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get('c', 3), 3) 5040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get('a'), 1) 5050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.get('a', 3), 1) 5060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_setdefault(self): 5080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_setdefault(self) 5090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 5100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.setdefault('key0') is None) 5110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.setdefault('key0', []) 5120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(d.setdefault('key0') is None) 5130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.setdefault('key', []).append(3) 5140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d['key'][0], 3) 5150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d.setdefault('key', []).append(4) 5160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(len(d['key']), 2) 5170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_popitem(self): 5190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_popitem(self) 5200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for copymode in -1, +1: 5210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # -1: b has same structure as a 5220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # +1: b is a.copy() 5230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for log2size in range(12): 5240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi size = 2**log2size 5250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi a = self._empty_mapping() 5260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi b = self._empty_mapping() 5270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for i in range(size): 5280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi a[repr(i)] = i 5290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if copymode < 0: 5300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi b[repr(i)] = i 5310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if copymode > 0: 5320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi b = a.copy() 5330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for i in range(size): 5340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi ka, va = ta = a.popitem() 5350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(va, int(ka)) 5360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi kb, vb = tb = b.popitem() 5370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(vb, int(kb)) 5380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not(copymode < 0 and ta != tb)) 5390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not a) 5400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not b) 5410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_pop(self): 5430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi BasicTestMappingProtocol.test_pop(self) 5440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # Tests for pop with specified key 5460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 5470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi k, v = 'abc', 'def' 5480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # verify longs/ints get same value when key > 32 bits (for 64-bit archs) 5500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi # see SF bug #689659 5510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = 4503599627370496L 5520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi y = 4503599627370496 5530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi h = self._full_mapping({x: 'anything', y: 'something else'}) 5540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(h[x], h[y]) 5550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(k, v), v) 5570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[k] = v 5580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(d.pop(k, 1), v) 5590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yiclass TestHashMappingProtocol(TestMappingProtocol): 5620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_getitem(self): 5640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi TestMappingProtocol.test_getitem(self) 5650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 5660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadEq(object): 5680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __eq__(self, other): 5690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 5700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __hash__(self): 5710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 24 5720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 5740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[BadEq()] = 42 5750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(KeyError, d.__getitem__, 23) 5760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadHash(object): 5780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi fail = False 5790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __hash__(self): 5800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if self.fail: 5810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 5820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi else: 5830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 42 5840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 5860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = BadHash() 5870c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[x] = 42 5880c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x.fail = True 5890c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.__getitem__, x) 5900c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 5910c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_fromkeys(self): 5920c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi TestMappingProtocol.test_fromkeys(self) 5930c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class mydict(self.type2test): 5940c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __new__(cls): 5950c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return UserDict.UserDict() 5960c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi ud = mydict.fromkeys('ab') 5970c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(ud, {'a':None, 'b':None}) 5980c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertIsInstance(ud, UserDict.UserDict) 5990c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6000c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_pop(self): 6010c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi TestMappingProtocol.test_pop(self) 6020c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6030c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 6040c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6050c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadHash(object): 6060c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi fail = False 6070c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __hash__(self): 6080c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if self.fail: 6090c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 6100c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi else: 6110c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 42 6120c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6130c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 6140c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = BadHash() 6150c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[x] = 42 6160c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x.fail = True 6170c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.pop, x) 6180c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6190c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_mutatingiteration(self): 6200c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 6210c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[1] = 1 6220c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi try: 6230c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi for i in d: 6240c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[i+1] = 1 6250c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi except RuntimeError: 6260c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi pass 6270c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi else: 6280c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.fail("changing dict size during iteration doesn't raise Error") 6290c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6300c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_repr(self): 6310c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 6320c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(repr(d), '{}') 6330c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[1] = 2 6340c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(repr(d), '{1: 2}') 6350c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 6360c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[1] = d 6370c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertEqual(repr(d), '{1: {...}}') 6380c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6390c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 6400c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6410c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadRepr(object): 6420c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __repr__(self): 6430c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 6440c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6450c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._full_mapping({1: BadRepr()}) 6460c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, repr, d) 6470c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6480c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_le(self): 6490c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not (self._empty_mapping() < self._empty_mapping())) 6500c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertTrue(not (self._full_mapping({1: 2}) < self._full_mapping({1L: 2L}))) 6510c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6520c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 6530c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6540c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadCmp(object): 6550c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __eq__(self, other): 6560c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 6570c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __hash__(self): 6580c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 42 6590c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6600c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d1 = self._full_mapping({BadCmp(): 1}) 6610c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d2 = self._full_mapping({1: 1}) 6620c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi try: 6630c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d1 < d2 6640c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi except Exc: 6650c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi pass 6660c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi else: 6670c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.fail("< didn't raise Exc") 6680c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6690c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def test_setdefault(self): 6700c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi TestMappingProtocol.test_setdefault(self) 6710c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6720c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class Exc(Exception): pass 6730c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6740c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi class BadHash(object): 6750c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi fail = False 6760c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi def __hash__(self): 6770c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi if self.fail: 6780c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi raise Exc() 6790c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi else: 6800c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi return 42 6810c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi 6820c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d = self._empty_mapping() 6830c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x = BadHash() 6840c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi d[x] = 42 6850c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi x.fail = True 6860c5958b1636c47ed7c284f859c8e805fd06a0e6Bill Yi self.assertRaises(Exc, d.setdefault, x, []) 687