summaryrefslogtreecommitdiff
path: root/coverage/bytecode.py
diff options
context:
space:
mode:
authorNed Batchelder <ned@nedbatchelder.com>2022-12-31 19:04:31 -0500
committerNed Batchelder <ned@nedbatchelder.com>2022-12-31 20:39:09 -0500
commite3c523c98cbc9ecdd37e36da19848ca9d0aef4a3 (patch)
tree0857c3472c5858695d26f0276e8f1c5b3512a3e2 /coverage/bytecode.py
parent5a72a1eb736516759201b223463f69f00979818e (diff)
downloadpython-coveragepy-git-e3c523c98cbc9ecdd37e36da19848ca9d0aef4a3.tar.gz
mypy: add __init__.py, __main__.py, bytecode.py, context.py, exceptions.py, report.py, and version.py
Diffstat (limited to 'coverage/bytecode.py')
-rw-r--r--coverage/bytecode.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/coverage/bytecode.py b/coverage/bytecode.py
index ceb18cf3..15bf755b 100644
--- a/coverage/bytecode.py
+++ b/coverage/bytecode.py
@@ -3,10 +3,11 @@
"""Bytecode manipulation for coverage.py"""
-import types
+from types import CodeType
+from typing import Generator
-def code_objects(code):
+def code_objects(code: CodeType) -> Generator[CodeType, None, None]:
"""Iterate over all the code objects in `code`."""
stack = [code]
while stack:
@@ -14,6 +15,6 @@ def code_objects(code):
# push its children for later returning.
code = stack.pop()
for c in code.co_consts:
- if isinstance(c, types.CodeType):
+ if isinstance(c, CodeType):
stack.append(c)
yield code