mirror of
https://github.com/OpenFreeEnergy/openfe.git
synced 2026-06-04 22:34:24 +08:00
e42432d5b0
* add pandas construction * removing csv writer * replicating original behavior * add rich table * split into function * make names clearer * adding adjustments from rebase * put it in a box * adding tsv flag * update tests to use tsv flag * add smoke test for pretty print * adding file creation test * adding news item --------- Co-authored-by: Irfan Alibay <IAlibay@users.noreply.github.com> Co-authored-by: Mike Henry <11765982+mikemhenry@users.noreply.github.com>
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
"""
|
|
Tests the easy start guide
|
|
|
|
- runs plan_rbfe_network with tyk2 inputs and checks the network created
|
|
- mocks the calculations and performs gathers on the mocked outputs
|
|
"""
|
|
import os
|
|
|
|
import pytest
|
|
from importlib import resources
|
|
from click.testing import CliRunner
|
|
from os import path
|
|
from unittest import mock
|
|
from openff.units import unit
|
|
|
|
from openfecli.commands.plan_rbfe_network import plan_rbfe_network
|
|
from openfecli.commands.quickrun import quickrun
|
|
from openfecli.commands.gather import gather
|
|
|
|
from .utils import assert_click_success
|
|
|
|
@pytest.fixture
|
|
def tyk2_ligands():
|
|
with resources.files('openfecli.tests.data.rbfe_tutorial') as d:
|
|
yield str(d / 'tyk2_ligands.sdf')
|
|
|
|
|
|
@pytest.fixture
|
|
def tyk2_protein():
|
|
with resources.files('openfecli.tests.data.rbfe_tutorial') as d:
|
|
yield str(d / 'tyk2_protein.pdb')
|
|
|
|
|
|
@pytest.fixture
|
|
def expected_transformations():
|
|
return [
|
|
"rbfe_lig_ejm_31_solvent_lig_ejm_48_solvent.json",
|
|
"rbfe_lig_ejm_46_solvent_lig_jmc_28_solvent.json",
|
|
"rbfe_lig_jmc_27_complex_lig_jmc_28_complex.json",
|
|
"rbfe_lig_jmc_23_solvent_lig_jmc_28_solvent.json",
|
|
"rbfe_lig_ejm_42_solvent_lig_ejm_50_solvent.json",
|
|
"rbfe_lig_ejm_31_complex_lig_ejm_46_complex.json",
|
|
"rbfe_lig_ejm_31_solvent_lig_ejm_50_solvent.json",
|
|
"rbfe_lig_ejm_42_solvent_lig_ejm_43_solvent.json",
|
|
"rbfe_lig_ejm_31_complex_lig_ejm_47_complex.json",
|
|
"rbfe_lig_jmc_27_solvent_lig_jmc_28_solvent.json",
|
|
"rbfe_lig_jmc_23_complex_lig_jmc_28_complex.json",
|
|
"rbfe_lig_ejm_42_complex_lig_ejm_50_complex.json",
|
|
"rbfe_lig_ejm_31_solvent_lig_ejm_46_solvent.json",
|
|
"rbfe_lig_ejm_31_complex_lig_ejm_50_complex.json",
|
|
"rbfe_lig_ejm_42_complex_lig_ejm_43_complex.json",
|
|
"rbfe_lig_ejm_31_solvent_lig_ejm_47_solvent.json",
|
|
"rbfe_lig_ejm_31_complex_lig_ejm_48_complex.json",
|
|
"rbfe_lig_ejm_46_complex_lig_jmc_28_complex.json",
|
|
]
|
|
|
|
|
|
def test_plan_tyk2(tyk2_ligands, tyk2_protein, expected_transformations):
|
|
runner = CliRunner()
|
|
|
|
with runner.isolated_filesystem():
|
|
result = runner.invoke(plan_rbfe_network, ['-M', tyk2_ligands,
|
|
'-p', tyk2_protein])
|
|
assert_click_success(result)
|
|
assert path.exists('alchemicalNetwork/transformations')
|
|
for f in expected_transformations:
|
|
assert path.exists(
|
|
path.join('alchemicalNetwork/transformations', f))
|
|
# make sure these are the only transforms
|
|
assert len(os.listdir("alchemicalNetwork/transformations")) == len(expected_transformations)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_execute(expected_transformations):
|
|
def fake_execute(*args, **kwargs):
|
|
return {
|
|
'repeat_id': kwargs['repeat_id'],
|
|
'generation': kwargs['generation'],
|
|
'nc': 'file.nc',
|
|
'last_checkpoint': 'checkpoint.nc',
|
|
'unit_estimate': 4.2 * unit.kilocalories_per_mole
|
|
}
|
|
|
|
with mock.patch('openfe.protocols.openmm_rfe.equil_rfe_methods.'
|
|
'RelativeHybridTopologyProtocolUnit._execute') as m:
|
|
m.side_effect = fake_execute
|
|
|
|
yield m
|
|
|
|
|
|
@pytest.fixture
|
|
def ref_gather():
|
|
return """\
|
|
ligand_i\tligand_j\tDDG(i->j) (kcal/mol)\tuncertainty (kcal/mol)
|
|
lig_ejm_31\tlig_ejm_46\t0.0\t0.0
|
|
lig_ejm_31\tlig_ejm_47\t0.0\t0.0
|
|
lig_ejm_31\tlig_ejm_48\t0.0\t0.0
|
|
lig_ejm_31\tlig_ejm_50\t0.0\t0.0
|
|
lig_ejm_42\tlig_ejm_43\t0.0\t0.0
|
|
lig_ejm_42\tlig_ejm_50\t0.0\t0.0
|
|
lig_ejm_46\tlig_jmc_28\t0.0\t0.0
|
|
lig_jmc_23\tlig_jmc_28\t0.0\t0.0
|
|
lig_jmc_27\tlig_jmc_28\t0.0\t0.0
|
|
"""
|
|
|
|
|
|
def test_run_tyk2(tyk2_ligands, tyk2_protein, expected_transformations,
|
|
mock_execute, ref_gather):
|
|
runner = CliRunner()
|
|
with runner.isolated_filesystem():
|
|
result = runner.invoke(plan_rbfe_network, ['-M', tyk2_ligands,
|
|
'-p', tyk2_protein])
|
|
|
|
assert_click_success(result)
|
|
|
|
for f in expected_transformations:
|
|
fn = path.join('alchemicalNetwork/transformations', f)
|
|
result2 = runner.invoke(quickrun, [fn])
|
|
assert_click_success(result2)
|
|
|
|
gather_result = runner.invoke(gather, ["--report", "ddg", '.', '--tsv'])
|
|
|
|
assert_click_success(gather_result)
|
|
assert gather_result.stdout == ref_gather
|