| 1234567891011121314151617 |
- from __future__ import annotations
- import os
- from unittest import TestCase, mock
- from mypy.util import get_terminal_width
- class TestGetTerminalSize(TestCase):
- def test_get_terminal_size_in_pty_defaults_to_80(self) -> None:
- # when run using a pty, `os.get_terminal_size()` returns `0, 0`
- ret = os.terminal_size((0, 0))
- mock_environ = os.environ.copy()
- mock_environ.pop("COLUMNS", None)
- with mock.patch.object(os, "get_terminal_size", return_value=ret):
- with mock.patch.dict(os.environ, values=mock_environ, clear=True):
- assert get_terminal_width() == 80
|
