diff options
-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 2a86c13..589d6cd 100644 --- a/test/test_main.py +++ b/test/test_main.py @@ -259,9 +259,10 @@ def test_reserved_space_is_integer(monkeypatch): def stub_terminal_size(): return (5, 5) - monkeypatch.setattr(shutil, 'get_terminal_size', stub_terminal_size) - mycli = MyCli() - assert isinstance(mycli.get_reserved_space(), int) + with monkeypatch.context() as m: + m.setattr(shutil, 'get_terminal_size', stub_terminal_size) + mycli = MyCli() + assert isinstance(mycli.get_reserved_space(), int) def test_list_dsn(): |