mirror of
https://github.com/KosinskiLab/AlphaPulldown.git
synced 2026-06-04 14:14:24 +08:00
fff63051b4
* Harden MMseqs species ID resolution fallback * Reorganize tests for CPU coverage CI * New * Fix function coverage checker def-line false positives * Expand unit coverage for helper and backend manager utilities * New. * New. * Expand unit coverage for template and post-processing helpers * Expand unit coverage for objects.py edge cases * Publish HTML coverage reports via GitHub Pages * Add CPU unit coverage for AlphaFold3 backend helpers * Reorganize tests and expand backend coverage * Reset shared test flags between cases * Expand AF3 prepare_input unit coverage * Cover AF3 and truemultimer feature creation * Test AF3 multimer MSA translation paths * Cover AF3 duplicate-residue multimer fallback * Cover AF2 resume and postprocess edge paths * Cover AF3 template mmCIF preparation * Test small script entry points * Expand workflow and ModelCIF test coverage * Add backend extras and install guide * Clarify AF3 backend installation path * Stabilize cluster GPU test runners * Document AF3 CMake SQLite hints * Simplify backend installation guide * Align AF3 install with working cluster env * Backfill typing dataclass_transform for AF2 * Pin TensorFlow for cluster installs * Fallback AF2 relax when CUDA OpenMM is unavailable * Raise AF3 default minimum bucket size * Simplify backend cluster installation guide * Fix AF3 wrapper JSON output isolation * Fix AF3 JSON wrapper outputs and MMseqs ID parsing * Fix CI entrypoint stub and Python 3.8 typing * Document release readiness test gates
23 lines
1.0 KiB
Python
23 lines
1.0 KiB
Python
import unittest
|
|
from alphapulldown.objects import MultimericObject
|
|
import pickle
|
|
from alphapulldown.folding_backend.alphalink_backend import AlphaLinkBackend
|
|
|
|
class TestAlphaLink2Backend(unittest.TestCase):
|
|
|
|
def setUp(self) -> None:
|
|
self.monomer1 = pickle.load(open("./test/test_data/H1134_A.pkl",'rb'))
|
|
self.monomer2 = pickle.load(open("./test/test_data/H1134_B.pkl",'rb'))
|
|
self.multimericObj = MultimericObject([self.monomer1,self.monomer2])
|
|
self.alphalink2_weights = '/g/alphafold/alphalink_weights/AlphaLink-Multimer_SDA_v3.pt'
|
|
self.xl_info = "./test/test_data/test_xl_input.pkl.gz"
|
|
return super().setUp()
|
|
|
|
def test_1_initialise_folding_backend(self):
|
|
beckend = AlphaLinkBackend
|
|
model_config = beckend.setup(self.alphalink2_weights,
|
|
self.xl_info)
|
|
beckend.predict(**model_config,multimeric_object = self.multimericObj,output_dir= "./test/test_data")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |