Skip to content
Snippets Groups Projects
test_run_engine.py 877 B
import unittest
import os

import ost
from promod3 import pipeline
from promod3 import rawmodel

class RunPipelineTests(unittest.TestCase):
    def testBuildFromRawModel(self):
        '''Positive test starting with a raw model.

        Should produce a real model.
        '''
        # preparing a tiny raw model
        template = ost.io.LoadPDB(os.path.join("data", "2l4k_B.pdb.gz"))
        aln = ost.io.LoadAlignment(os.path.join("data", "2l4k_B_aln.fasta"))
        aln.AttachView(1, template.Select('peptide=true'))
        raw_model = rawmodel.BuildRawModel(aln)
        # run the pipeline
        self.assertTrue(pipeline.run_engine.BuildFromRawModel(raw_model))

if __name__ == "__main__":
    from ost import testutils
    testutils.RunTests()

#  LocalWords:  unittest promod RunPipelineTests TestCase ost testutils
#  LocalWords:  testBuildFromRawModel RunTests