Fix mocker usage

2019-12-04 22:46:01 +07:00 · 2019-12-04 22:46:01 +07:00 · b21545348e
parent 2e4e5036dd
commit b21545348e
1 changed files with 3 additions and 4 deletions
--- a/test/test_version.py
+++ b/test/test_version.py
@ -3,7 +3,6 @@

 def test_version_fails_gracefully(mocker):
    target = 'pkg_resources.get_distribution'
-    with mocker.patch(target, side_effect=Exception()):
-        from flat_tree.__init__ import __version__
-
-        assert __version__ == 'unknown'
+    mocker.patch(target, side_effect=Exception())
+    from flat_tree.__init__ import __version__
+    assert __version__ == 'unknown'