18 lines
629 B
Python
18 lines
629 B
Python
|
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
|