30 lines
1.0 KiB
Diff
30 lines
1.0 KiB
Diff
--- a/tests/unit/test_venv.py
|
|
+++ b/tests/unit/test_venv.py
|
|
@@ -1,4 +1,7 @@
|
|
-import mock
|
|
+try:
|
|
+ from unittest.mock import patch
|
|
+except ImportError:
|
|
+ from mock import patch
|
|
|
|
import pytest_virtualenv as venv
|
|
from pytest_shutil import env
|
|
@@ -6,7 +9,7 @@ from pytest_shutil import env
|
|
|
|
def test_PYTHONPATH_not_present_in_testing_env_if_set():
|
|
with env.set_env('PYTHONPATH', 'fred'):
|
|
- with mock.patch.object(venv.Workspace, 'run') as run:
|
|
+ with patch.object(venv.Workspace, 'run') as run:
|
|
venv.VirtualEnv()
|
|
call = run.mock_calls[0]
|
|
assert 'PYTHONPATH' not in call[2]['env']
|
|
@@ -18,7 +21,7 @@ def test_PYTHONPATH_not_present_in_testi
|
|
|
|
def test_PYTHONPATH_not_present_in_testing_env_if_unset():
|
|
with env.no_env('PYTHONPATH'):
|
|
- with mock.patch.object(venv.Workspace, 'run') as run:
|
|
+ with patch.object(venv.Workspace, 'run') as run:
|
|
venv.VirtualEnv()
|
|
call = run.mock_calls[0]
|
|
assert 'PYTHONPATH' not in call[2]['env']
|