[Python-checkins] r53656 - in python/trunk: Lib/test/test_dict.py Objects/dictobject.c

[raymond.hettinger]

Author: raymond.hettinger
Date: Wed Feb  7 21:08:22 2007
New Revision: 53656

Modified:
   python/trunk/Lib/test/test_dict.py
   python/trunk/Objects/dictobject.c
Log:
SF #1615701:  make d.update(m) honor __getitem__() and keys() in dict subclasses



Modified: python/trunk/Lib/test/test_dict.py
==============================================================================
--- python/trunk/Lib/test/test_dict.py	(original)
+++ python/trunk/Lib/test/test_dict.py	Wed Feb  7 21:08:22 2007
@@ -189,6 +189,14 @@
 
         self.assertRaises(ValueError, {}.update, [(1, 2, 3)])
 
+        # SF #1615701:  make d.update(m) honor __getitem__() and keys() in dict subclasses
+        class KeyUpperDict(dict):
+            def __getitem__(self, key):
+                return key.upper()
+        d.clear()
+        d.update(KeyUpperDict.fromkeys('abc'))
+        self.assertEqual(d, {'a':'A', 'b':'B', 'c':'C'})
+
     def test_fromkeys(self):
         self.assertEqual(dict.fromkeys('abc'), {'a':None, 'b':None, 'c':None})
         d = {}

Modified: python/trunk/Objects/dictobject.c
==============================================================================
--- python/trunk/Objects/dictobject.c	(original)
+++ python/trunk/Objects/dictobject.c	Wed Feb  7 21:08:22 2007
@@ -1306,7 +1306,7 @@
 		return -1;
 	}
 	mp = (dictobject*)a;
-	if (PyDict_Check(b)) {
+	if (PyDict_CheckExact(b)) {
 		other = (dictobject*)b;
 		if (other == mp || other->ma_used == 0)
 			/* a.update(a) or a.update({}); nothing to do */