From 414a87a53f1b424c5e5073583e6cd978857a1d9b Mon Sep 17 00:00:00 2001 From: Ran Benita Date: Thu, 30 Apr 2020 14:20:48 +0300 Subject: [PATCH] config/argparsing: use our own get_terminal_width() --- src/_pytest/_io/__init__.py | 2 ++ src/_pytest/config/argparsing.py | 3 ++- testing/test_config.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/_pytest/_io/__init__.py b/src/_pytest/_io/__init__.py index 880c3c87a..db001e918 100644 --- a/src/_pytest/_io/__init__.py +++ b/src/_pytest/_io/__init__.py @@ -1,6 +1,8 @@ +from .terminalwriter import get_terminal_width from .terminalwriter import TerminalWriter __all__ = [ "TerminalWriter", + "get_terminal_width", ] diff --git a/src/_pytest/config/argparsing.py b/src/_pytest/config/argparsing.py index 140e04e97..940eaa6a7 100644 --- a/src/_pytest/config/argparsing.py +++ b/src/_pytest/config/argparsing.py @@ -15,6 +15,7 @@ from typing import Union import py +import _pytest._io from _pytest.compat import TYPE_CHECKING from _pytest.config.exceptions import UsageError @@ -466,7 +467,7 @@ class DropShorterLongHelpFormatter(argparse.HelpFormatter): def __init__(self, *args: Any, **kwargs: Any) -> None: """Use more accurate terminal width via pylib.""" if "width" not in kwargs: - kwargs["width"] = py.io.get_terminal_width() + kwargs["width"] = _pytest._io.get_terminal_width() super().__init__(*args, **kwargs) def _format_action_invocation(self, action: argparse.Action) -> str: diff --git a/testing/test_config.py b/testing/test_config.py index 9035407b7..0c05c4fad 100644 --- a/testing/test_config.py +++ b/testing/test_config.py @@ -1253,7 +1253,7 @@ def test_help_formatter_uses_py_get_terminal_width(monkeypatch): formatter = DropShorterLongHelpFormatter("prog") assert formatter._width == 90 - monkeypatch.setattr("py.io.get_terminal_width", lambda: 160) + monkeypatch.setattr("_pytest._io.get_terminal_width", lambda: 160) formatter = DropShorterLongHelpFormatter("prog") assert formatter._width == 160