Source code for schrodinger.application.desmond.test_utils

import tarfile
from typing import Tuple
from schrodinger.test import mmshare_testfile


def unzip_and_get_membrane_test_files(fep_type: str) -> Tuple[str, str, str]:
    """
    Unzip test files for the membrane launcher workflow into cwd and return
    filenames. The first file is the expected .fmp input file for the
    membrane launcher stage, the second file is the expected .cms output
    file from the multisim job and the third file is the expected .fmp output
    file from the stage.
    """
    tar_fname = mmshare_testfile(f"desmond/auto_membrane/{fep_type}.tar.gz")
    with tarfile.open(tar_fname) as tar_handle:
        tar_handle.extractall()
    return ('membrane_relaxation_2.fmp', 'relaxed.cms',
            'membrane_relaxation_3.fmp')