diff options
Diffstat (limited to 'tests/test_main.py')
| -rw-r--r-- | tests/test_main.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/test_main.py b/tests/test_main.py index 90132f1..d4d52af 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -2,6 +2,7 @@ import os from collections import namedtuple from textwrap import dedent from tempfile import NamedTemporaryFile +import shutil import click from click.testing import CliRunner @@ -232,12 +233,12 @@ def test_reserved_space_is_integer(): def stub_terminal_size(): return (5, 5) - old_func = click.get_terminal_size + old_func = shutil.get_terminal_size - click.get_terminal_size = stub_terminal_size + shutil.get_terminal_size = stub_terminal_size lc = LiteCli() assert isinstance(lc.get_reserved_space(), int) - click.get_terminal_size = old_func + shutil.get_terminal_size = old_func @dbtest |