mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-10 07:41:51 +08:00
Try to use the virtual env that was defined in the VIRTUAL_ENV variable, if possible.
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user