diff options
Diffstat (limited to 'tests/plugin_config.py')
-rw-r--r-- | tests/plugin_config.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/plugin_config.py b/tests/plugin_config.py index c4d3cf65..bb6893e3 100644 --- a/tests/plugin_config.py +++ b/tests/plugin_config.py @@ -3,20 +3,29 @@ """A configuring plugin for test_plugins.py to import.""" +from __future__ import annotations + +from typing import Any, List, cast + import coverage +from coverage.plugin_support import Plugins +from coverage.types import TConfigurable class Plugin(coverage.CoveragePlugin): """A configuring plugin for testing.""" - def configure(self, config): + def configure(self, config: TConfigurable) -> None: """Configure all the things!""" opt_name = "report:exclude_lines" - exclude_lines = config.get_option(opt_name) + exclude_lines = cast(List[str], config.get_option(opt_name)) exclude_lines.append(r"pragma: custom") exclude_lines.append(r"pragma: or whatever") config.set_option(opt_name, exclude_lines) -def coverage_init(reg, options): # pylint: disable=unused-argument +def coverage_init( + reg: Plugins, + options: Any, # pylint: disable=unused-argument +) -> None: """Called by coverage to initialize the plugins here.""" reg.add_configurer(Plugin()) |