1
0
forked from VimPlug/jedi

Try to use the virtual env that was defined in the VIRTUAL_ENV variable, if possible.

This commit is contained in:
Dave Halter
2018-01-02 01:28:02 +01:00
parent bcb3f02a01
commit 9c5ce5a8d2
2 changed files with 28 additions and 0 deletions

View File

@@ -1,3 +1,7 @@
import os
import sys
from contextlib import contextmanager
import pytest
import jedi
@@ -71,3 +75,21 @@ def test_killed_subprocess(evaluator, Script):
def_, = Script('str').goto_definitions()
# Jedi should now work again.
assert def_.name == 'str'
@contextmanager
def set_environment_variable(name, value):
tmp = os.environ.get(name)
try:
os.environ[name] = value
yield
finally:
if tmp is None:
del os.environ[name]
else:
os.environ[name] = tmp
def test_virtualenv():
with set_environment_variable('VIRTUAL_ENV', '/foo/bar/jedi_baz'):
assert get_default_environment()._executable == sys.executable