Extend work on rev 52962 and 53829 eliminating redundant PyObject_Hash() calls and fixing set/dict interoperability.
This commit is contained in:
parent
ce55e21c70
commit
0bbbfc4c0b
|
@ -82,7 +82,8 @@ PyAPI_FUNC(int) PySet_Clear(PyObject *set);
|
||||||
PyAPI_FUNC(int) PySet_Contains(PyObject *anyset, PyObject *key);
|
PyAPI_FUNC(int) PySet_Contains(PyObject *anyset, PyObject *key);
|
||||||
PyAPI_FUNC(int) PySet_Discard(PyObject *set, PyObject *key);
|
PyAPI_FUNC(int) PySet_Discard(PyObject *set, PyObject *key);
|
||||||
PyAPI_FUNC(int) PySet_Add(PyObject *set, PyObject *key);
|
PyAPI_FUNC(int) PySet_Add(PyObject *set, PyObject *key);
|
||||||
PyAPI_FUNC(int) _PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **entry);
|
PyAPI_FUNC(int) _PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **key);
|
||||||
|
PyAPI_FUNC(int) _PySet_NextEntry(PyObject *set, Py_ssize_t *pos, PyObject **key, long *hash);
|
||||||
PyAPI_FUNC(PyObject *) PySet_Pop(PyObject *set);
|
PyAPI_FUNC(PyObject *) PySet_Pop(PyObject *set);
|
||||||
PyAPI_FUNC(int) _PySet_Update(PyObject *set, PyObject *iterable);
|
PyAPI_FUNC(int) _PySet_Update(PyObject *set, PyObject *iterable);
|
||||||
|
|
||||||
|
|
|
@ -289,6 +289,10 @@ class TestJointOps(unittest.TestCase):
|
||||||
if hasattr(s, 'symmetric_difference_update'):
|
if hasattr(s, 'symmetric_difference_update'):
|
||||||
s.symmetric_difference_update(d)
|
s.symmetric_difference_update(d)
|
||||||
self.assertEqual(sum(elem.hash_count for elem in d), n)
|
self.assertEqual(sum(elem.hash_count for elem in d), n)
|
||||||
|
d2 = dict.fromkeys(set(d))
|
||||||
|
self.assertEqual(sum(elem.hash_count for elem in d), n)
|
||||||
|
d3 = dict.fromkeys(frozenset(d))
|
||||||
|
self.assertEqual(sum(elem.hash_count for elem in d), n)
|
||||||
|
|
||||||
class TestSet(TestJointOps):
|
class TestSet(TestJointOps):
|
||||||
thetype = set
|
thetype = set
|
||||||
|
|
|
@ -1175,6 +1175,24 @@ dict_fromkeys(PyObject *cls, PyObject *args)
|
||||||
if (d == NULL)
|
if (d == NULL)
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
|
if (PyDict_CheckExact(d) && PyAnySet_CheckExact(seq)) {
|
||||||
|
dictobject *mp = (dictobject *)d;
|
||||||
|
Py_ssize_t pos = 0;
|
||||||
|
PyObject *key;
|
||||||
|
long hash;
|
||||||
|
|
||||||
|
if (dictresize(mp, PySet_GET_SIZE(seq)))
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
while (_PySet_NextEntry(seq, &pos, &key, &hash)) {
|
||||||
|
Py_INCREF(key);
|
||||||
|
Py_INCREF(Py_None);
|
||||||
|
if (insertdict(mp, key, hash, Py_None))
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return d;
|
||||||
|
}
|
||||||
|
|
||||||
it = PyObject_GetIter(seq);
|
it = PyObject_GetIter(seq);
|
||||||
if (it == NULL){
|
if (it == NULL){
|
||||||
Py_DECREF(d);
|
Py_DECREF(d);
|
||||||
|
|
|
@ -2137,7 +2137,7 @@ PySet_Add(PyObject *set, PyObject *key)
|
||||||
}
|
}
|
||||||
|
|
||||||
int
|
int
|
||||||
_PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **entry)
|
_PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **key)
|
||||||
{
|
{
|
||||||
setentry *entry_ptr;
|
setentry *entry_ptr;
|
||||||
|
|
||||||
|
@ -2147,7 +2147,23 @@ _PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **entry)
|
||||||
}
|
}
|
||||||
if (set_next((PySetObject *)set, pos, &entry_ptr) == 0)
|
if (set_next((PySetObject *)set, pos, &entry_ptr) == 0)
|
||||||
return 0;
|
return 0;
|
||||||
*entry = entry_ptr->key;
|
*key = entry_ptr->key;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
_PySet_NextEntry(PyObject *set, Py_ssize_t *pos, PyObject **key, long *hash)
|
||||||
|
{
|
||||||
|
setentry *entry;
|
||||||
|
|
||||||
|
if (!PyAnySet_Check(set)) {
|
||||||
|
PyErr_BadInternalCall();
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (set_next((PySetObject *)set, pos, &entry) == 0)
|
||||||
|
return 0;
|
||||||
|
*key = entry->key;
|
||||||
|
*hash = entry->hash;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue