|
|
"""Test path helpers.""" |
|
|
|
|
|
from pathlib import Path |
|
|
|
|
|
import pytest |
|
|
import typer |
|
|
from folding_studio.utils.input_validation import ( |
|
|
extract_and_validate_custom_msas, |
|
|
extract_and_validate_custom_templates, |
|
|
validate_initial_guess, |
|
|
) |
|
|
|
|
|
|
|
|
def test_extract_and_validate_custom_templates_pass(tmp_path: Path): |
|
|
"""Test extract and validate custom templates pass.""" |
|
|
|
|
|
test_str_paths = [ |
|
|
tmp_path / "template.cif", |
|
|
tmp_path / "dir_1/template_1.cif", |
|
|
tmp_path / "dir_1/template_2.cif", |
|
|
tmp_path / "dir_1/dir_11/template.cif", |
|
|
tmp_path / "dir_2/template_1.cif", |
|
|
tmp_path / "dir_2/template_2.cif", |
|
|
tmp_path / "dir_3/template_1.cif", |
|
|
tmp_path / "dir_3/template_2.cif", |
|
|
] |
|
|
for path in test_str_paths: |
|
|
path.parent.mkdir(exist_ok=True, parents=True) |
|
|
path.touch() |
|
|
|
|
|
test_paths = [ |
|
|
tmp_path / "template.cif", |
|
|
tmp_path / "dir_1", |
|
|
tmp_path / "dir_2", |
|
|
tmp_path / "dir_3/template_1.cif", |
|
|
] |
|
|
|
|
|
extracted_paths = extract_and_validate_custom_templates(test_paths) |
|
|
|
|
|
assert len(extracted_paths) == 6 |
|
|
|
|
|
|
|
|
def test_extract_and_validate_custom_templates_fails_if_unsupported(tmp_path: Path): |
|
|
"""Test extract and validate custom templates fails if an unsupported files is passed.""" |
|
|
file = tmp_path / "template.txt" |
|
|
file.touch() |
|
|
test_paths = [Path(file)] |
|
|
with pytest.raises( |
|
|
typer.BadParameter, match=f"The file '{file}' is not supported." |
|
|
): |
|
|
extract_and_validate_custom_templates(test_paths) |
|
|
|
|
|
|
|
|
def test_extract_and_validate_custom_msass_pass(tmp_path: Path): |
|
|
"""Test extract and validate custom msas pass.""" |
|
|
|
|
|
test_str_paths = [ |
|
|
tmp_path / "msa.sto", |
|
|
tmp_path / "dir_1/msa_1.sto", |
|
|
tmp_path / "dir_1/msa_2.a3m", |
|
|
tmp_path / "dir_1/dir_11/msa.sto", |
|
|
tmp_path / "dir_2/msa_1.sto", |
|
|
tmp_path / "dir_2/msa_2.a3m", |
|
|
tmp_path / "dir_3/msa_1.sto", |
|
|
tmp_path / "dir_3/msa_2.a3m", |
|
|
] |
|
|
for path in test_str_paths: |
|
|
path.parent.mkdir(exist_ok=True, parents=True) |
|
|
path.touch() |
|
|
|
|
|
test_paths = [ |
|
|
tmp_path / "msa.sto", |
|
|
tmp_path / "dir_1", |
|
|
tmp_path / "dir_2", |
|
|
tmp_path / "dir_3/msa_1.sto", |
|
|
] |
|
|
|
|
|
extracted_paths = extract_and_validate_custom_msas(test_paths) |
|
|
|
|
|
assert len(extracted_paths) == 6 |
|
|
|
|
|
|
|
|
def test_extract_and_validate_custom_msas_fails_if_unsupported(tmp_path: Path): |
|
|
"""Test extract and validate custom msas fails if an unsupported file is passed.""" |
|
|
file = tmp_path / "msa.txt" |
|
|
file.touch() |
|
|
test_paths = [Path(file)] |
|
|
with pytest.raises( |
|
|
typer.BadParameter, match=f"The file '{file}' is not supported." |
|
|
): |
|
|
extract_and_validate_custom_templates(test_paths) |
|
|
|
|
|
|
|
|
@pytest.mark.parametrize( |
|
|
("file", "expected_path"), |
|
|
[ |
|
|
(Path("initial_guess.cif"), Path("initial_guess.cif")), |
|
|
(Path("dir_1/initial_guess_1.cif"), Path("dir_1/initial_guess_1.cif")), |
|
|
(None, None), |
|
|
], |
|
|
) |
|
|
def test_extract_and_validate_initial_guess_path(file: str, expected_path): |
|
|
"""Test extract and validate initial guess file path.""" |
|
|
|
|
|
validated_path = validate_initial_guess(file) |
|
|
|
|
|
assert validated_path == expected_path |
|
|
|
