|
50 | 50 | from _pytest.nodes import Collector
|
51 | 51 | from _pytest.nodes import Item
|
52 | 52 | from _pytest.pathlib import make_numbered_dir
|
53 |
| -from _pytest.python import Module |
54 | 53 | from _pytest.reports import CollectReport
|
55 | 54 | from _pytest.reports import TestReport
|
56 | 55 | from _pytest.tmpdir import TempPathFactory
|
@@ -652,7 +651,7 @@ def __init__(
|
652 | 651 | ) -> None:
|
653 | 652 | self._request = request
|
654 | 653 | self._mod_collections: WeakKeyDictionary[
|
655 |
| - Module, List[Union[Item, Collector]] |
| 654 | + Collector, List[Union[Item, Collector]] |
656 | 655 | ] = (WeakKeyDictionary())
|
657 | 656 | if request.function:
|
658 | 657 | name: str = request.function.__name__
|
@@ -1244,7 +1243,7 @@ def getmodulecol(
|
1244 | 1243 | return self.getnode(config, path)
|
1245 | 1244 |
|
1246 | 1245 | def collect_by_name(
|
1247 |
| - self, modcol: Module, name: str |
| 1246 | + self, modcol: Collector, name: str |
1248 | 1247 | ) -> Optional[Union[Item, Collector]]:
|
1249 | 1248 | """Return the collection node for name from the module collection.
|
1250 | 1249 |
|
@@ -1639,7 +1638,7 @@ def getmodulecol(self, source, configargs=(), withinit=False):
|
1639 | 1638 | )
|
1640 | 1639 |
|
1641 | 1640 | def collect_by_name(
|
1642 |
| - self, modcol: Module, name: str |
| 1641 | + self, modcol: Collector, name: str |
1643 | 1642 | ) -> Optional[Union[Item, Collector]]:
|
1644 | 1643 | """See :meth:`Pytester.collect_by_name`."""
|
1645 | 1644 | return self._pytester.collect_by_name(modcol, name)
|
|
0 commit comments