diff options
Diffstat (limited to 'test/test_main.py')
| -rw-r--r-- | test/test_main.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/test/test_main.py b/test/test_main.py index 00fdc1b..7731603 100644 --- a/test/test_main.py +++ b/test/test_main.py @@ -1,4 +1,5 @@ import os +import shutil import click from click.testing import CliRunner @@ -258,13 +259,13 @@ 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 mycli = MyCli() assert isinstance(mycli.get_reserved_space(), int) - click.get_terminal_size = old_func + shutil.get_terminal_size = old_func def test_list_dsn(): |