summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/test/test_genericalias.py4
-rw-r--r--Modules/itertoolsmodule.c2
2 files changed, 5 insertions, 1 deletions
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 */
};