From a8403d057d41a022d819fd8d6fbe2a666f72b6f5 Mon Sep 17 00:00:00 2001 From: Ethan Smith Date: Thu, 9 Apr 2020 20:28:08 -0700 Subject: Generic itertools.chain (GH-19417) --- Lib/test/test_genericalias.py | 4 +++- Modules/itertoolsmodule.c | 2 ++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/Lib/test/test_genericalias.py b/Lib/test/test_genericalias.py index 9c5b23e..d8acdee 100644 --- a/Lib/test/test_genericalias.py +++ b/Lib/test/test_genericalias.py @@ -7,6 +7,7 @@ from collections import ( ) from collections.abc import * from contextlib import AbstractContextManager, AbstractAsyncContextManager +from itertools import chain from os import DirEntry from re import Pattern, Match from types import GenericAlias, MappingProxyType @@ -35,7 +36,8 @@ class BaseTest(unittest.TestCase): Mapping, MutableMapping, MappingView, KeysView, ItemsView, ValuesView, Sequence, MutableSequence, - MappingProxyType, DirEntry + MappingProxyType, DirEntry, + chain, ): tname = t.__name__ with self.subTest(f"Testing {tname}"): diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index e99c964..4ebeb74 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -2028,6 +2028,8 @@ static PyMethodDef chain_methods[] = { reduce_doc}, {"__setstate__", (PyCFunction)chain_setstate, METH_O, setstate_doc}, + {"__class_getitem__", (PyCFunction)Py_GenericAlias, + METH_O|METH_CLASS, PyDoc_STR("See PEP 585")}, {NULL, NULL} /* sentinel */ }; -- cgit v0.12