chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
import sys
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
|
||||
from ray._private.runtime_env import uv
|
||||
|
||||
|
||||
class TestRuntimeEnv:
|
||||
def uv_config(self):
|
||||
return {"packages": ["requests"]}
|
||||
|
||||
def env_vars(self):
|
||||
return {}
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_install_uv():
|
||||
with patch(
|
||||
"ray._private.runtime_env.uv.UvProcessor._install_uv"
|
||||
) as mock_install_uv:
|
||||
mock_install_uv.return_value = None
|
||||
yield mock_install_uv
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_install_uv_packages():
|
||||
with patch(
|
||||
"ray._private.runtime_env.uv.UvProcessor._install_uv_packages"
|
||||
) as mock_install_uv_packages:
|
||||
mock_install_uv_packages.return_value = None
|
||||
yield mock_install_uv_packages
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run(mock_install_uv, mock_install_uv_packages):
|
||||
target_dir = "/tmp"
|
||||
runtime_env = TestRuntimeEnv()
|
||||
|
||||
uv_processor = uv.UvProcessor(target_dir=target_dir, runtime_env=runtime_env)
|
||||
await uv_processor._run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(pytest.main(["-vv", __file__]))
|
||||
Reference in New Issue
Block a user