From b46ad5431d2643f61e929c1ffec48766b2fafd75 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Fri, 21 Sep 2018 01:46:41 -0700 Subject: [PATCH] Minor performance tweak for deque.index() with a start argument (GH-9440) --- Lib/test/test_deque.py | 8 ++++++++ Modules/_collectionsmodule.c | 6 ++++-- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/Lib/test/test_deque.py b/Lib/test/test_deque.py index 921136069d7..51b66b76aca 100644 --- a/Lib/test/test_deque.py +++ b/Lib/test/test_deque.py @@ -288,6 +288,14 @@ class TestBasic(unittest.TestCase): else: self.assertEqual(d.index(element, start, stop), target) + # Test large start argument + d = deque(range(0, 10000, 10)) + for step in range(100): + i = d.index(8500, 700) + self.assertEqual(d[i], 8500) + # Repeat test with a different internal offset + d.rotate() + def test_index_bug_24913(self): d = deque('A' * 3) with self.assertRaises(ValueError): diff --git a/Modules/_collectionsmodule.c b/Modules/_collectionsmodule.c index 935b4348a8f..267cf07f1f7 100644 --- a/Modules/_collectionsmodule.c +++ b/Modules/_collectionsmodule.c @@ -1050,8 +1050,10 @@ deque_index(dequeobject *deque, PyObject *const *args, Py_ssize_t nargs) start = stop; assert(0 <= start && start <= stop && stop <= Py_SIZE(deque)); - /* XXX Replace this loop with faster code from deque_item() */ - for (i=0 ; irightlink; + } + for ( ; i < start ; i++) { index++; if (index == BLOCKLEN) { b = b->rightlink;