Issue #24379: Add operator.subscript() as a convenience for building slices.

This commit is contained in:
Raymond Hettinger 2015-08-16 19:43:34 -07:00
parent 755cb0ae8d
commit 47d159f522
4 changed files with 81 additions and 2 deletions

View File

@ -333,6 +333,21 @@ expect a function argument.
[('orange', 1), ('banana', 2), ('apple', 3), ('pear', 5)]
.. data:: subscript
A helper to turn subscript notation into indexing objects. This can be
used to create item access patterns ahead of time to pass them into
various subscriptable objects.
For example:
* ``subscript[5] == 5``
* ``subscript[3:7:2] == slice(3, 7, 2)``
* ``subscript[5, 8] == (5, 8)``
.. versionadded:: 3.6
.. function:: methodcaller(name[, args...])
Return a callable object that calls the method *name* on its operand. If

View File

@ -95,7 +95,12 @@ New Modules
Improved Modules
================
* None yet.
operator
--------
* New object :data:`operator.subscript` makes it easier to create complex
indexers. For example: ``subscript[0:10:2] == slice(0, 10, 2)``
(Contributed by Joe Jevnik in :issue:`24379`.)
Optimizations

View File

@ -17,7 +17,7 @@ __all__ = ['abs', 'add', 'and_', 'attrgetter', 'concat', 'contains', 'countOf',
'is_', 'is_not', 'isub', 'itemgetter', 'itruediv', 'ixor', 'le',
'length_hint', 'lshift', 'lt', 'matmul', 'methodcaller', 'mod',
'mul', 'ne', 'neg', 'not_', 'or_', 'pos', 'pow', 'rshift',
'setitem', 'sub', 'truediv', 'truth', 'xor']
'setitem', 'sub', 'subscript', 'truediv', 'truth', 'xor']
from builtins import abs as _abs
@ -408,6 +408,32 @@ def ixor(a, b):
return a
@object.__new__ # create a singleton instance
class subscript:
"""
A helper to turn subscript notation into indexing objects. This can be
used to create item access patterns ahead of time to pass them into
various subscriptable objects.
For example:
subscript[5] == 5
subscript[3:7:2] == slice(3, 7, 2)
subscript[5, 8] == (5, 8)
"""
__slots__ = ()
def __new__(cls):
raise TypeError("cannot create '{}' instances".format(cls.__name__))
@staticmethod
def __getitem__(key):
return key
@staticmethod
def __reduce__():
return 'subscript'
try:
from _operator import *
except ImportError:

View File

@ -596,5 +596,38 @@ class CCOperatorPickleTestCase(OperatorPickleTestCase, unittest.TestCase):
module2 = c_operator
class SubscriptTestCase:
def test_subscript(self):
subscript = self.module.subscript
self.assertIsNone(subscript[None])
self.assertEqual(subscript[0], 0)
self.assertEqual(subscript[0:1:2], slice(0, 1, 2))
self.assertEqual(
subscript[0, ..., :2, ...],
(0, Ellipsis, slice(2), Ellipsis),
)
def test_pickle(self):
from operator import subscript
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
with self.subTest(proto=proto):
self.assertIs(
pickle.loads(pickle.dumps(subscript, proto)),
subscript,
)
def test_singleton(self):
with self.assertRaises(TypeError):
type(self.module.subscript)()
def test_immutable(self):
with self.assertRaises(AttributeError):
self.module.subscript.attr = None
class PySubscriptTestCase(SubscriptTestCase, PyOperatorTestCase):
pass
if __name__ == "__main__":
unittest.main()