diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2022-12-31 19:04:31 -0500 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2022-12-31 20:39:09 -0500 |
commit | e3c523c98cbc9ecdd37e36da19848ca9d0aef4a3 (patch) | |
tree | 0857c3472c5858695d26f0276e8f1c5b3512a3e2 /coverage/bytecode.py | |
parent | 5a72a1eb736516759201b223463f69f00979818e (diff) | |
download | python-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.py | 7 |
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 |