72 lines
1.7 KiB
Python
72 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os.path
|
|
|
|
import pytest
|
|
|
|
from pre_commit import envcontext
|
|
from pre_commit.languages import conda
|
|
from pre_commit.store import _make_local_repo
|
|
from testing.language_helpers import run_language
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('ctx', 'expected'),
|
|
(
|
|
pytest.param(
|
|
(
|
|
('PRE_COMMIT_USE_MICROMAMBA', envcontext.UNSET),
|
|
('PRE_COMMIT_USE_MAMBA', envcontext.UNSET),
|
|
),
|
|
'conda',
|
|
id='default',
|
|
),
|
|
pytest.param(
|
|
(
|
|
('PRE_COMMIT_USE_MICROMAMBA', '1'),
|
|
('PRE_COMMIT_USE_MAMBA', ''),
|
|
),
|
|
'micromamba',
|
|
id='default',
|
|
),
|
|
pytest.param(
|
|
(
|
|
('PRE_COMMIT_USE_MICROMAMBA', ''),
|
|
('PRE_COMMIT_USE_MAMBA', '1'),
|
|
),
|
|
'mamba',
|
|
id='default',
|
|
),
|
|
),
|
|
)
|
|
def test_conda_exe(ctx, expected):
|
|
with envcontext.envcontext(ctx):
|
|
assert conda._conda_exe() == expected
|
|
|
|
|
|
def test_conda_language(tmp_path):
|
|
environment_yml = '''\
|
|
channels: [conda-forge, defaults]
|
|
dependencies: [python, pip]
|
|
'''
|
|
tmp_path.joinpath('environment.yml').write_text(environment_yml)
|
|
|
|
ret, out = run_language(
|
|
tmp_path,
|
|
conda,
|
|
'python -c "import sys; print(sys.prefix)"',
|
|
)
|
|
assert ret == 0
|
|
assert os.path.basename(out.strip()) == b'conda-default'
|
|
|
|
|
|
def test_conda_additional_deps(tmp_path):
|
|
_make_local_repo(tmp_path)
|
|
|
|
ret = run_language(
|
|
tmp_path,
|
|
conda,
|
|
'python -c "import botocore; print(1)"',
|
|
deps=('botocore',),
|
|
)
|
|
assert ret == (0, b'1\n')
|