def test_in_virtualenv(start_cluster):
assert (
PipProcessor._is_in_virtualenv() is False and "IN_VIRTUALENV" not in os.environ
) or (PipProcessor._is_in_virtualenv() is True and "IN_VIRTUALENV" in os.environ)
cluster, address = start_cluster
runtime_env = {"pip": ["pip-install-test==0.5"]}
ray.init(address, runtime_env=runtime_env)
@ray.remote
def f():
import pip_install_test # noqa: F401
return PipProcessor._is_in_virtualenv()
# Ensure that the runtime env has been installed
# and virtualenv is activated.
assert ray.get(f.remote())
def cloned_virtualenv():
# Lazy import pytest_virtualenv,
# aviod import `pytest_virtualenv` in test case `Minimal install`
from pytest_virtualenv import VirtualEnv
if PipProcessor._is_in_virtualenv():
raise RuntimeError("Forbid the use of this fixture in virtualenv")
venv = VirtualEnv(args=[
"--system-site-packages",
"--reset-app-data",
"--no-periodic-update",
"--no-download",
], )
yield venv
venv.teardown()
def f():
import pip_install_test # noqa: F401
return PipProcessor._is_in_virtualenv()