features_test_regions.py

import unittest
from abpytools.features.regions import ChainDomains
from abpytools.core.chain_collection import ChainCollection


class AbDomainTest(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.ab_file = './tests/Data/chain_collection_heavy_2_sequences.json'

    def test_ChainDomains_instantiation(self):
        chain_domain = ChainDomains(path=self.ab_file)
        self.assertIsInstance(chain_domain, ChainDomains)

    def test_ChainDomains_ChainCollection_instantiation(self):
        chain_collection = ChainCollection.load_from_file(self.ab_file)
        chain_domain = ChainDomains(antibody_objects=chain_collection)
        self.assertIsInstance(chain_domain, ChainDomains)

    def test_ChainDomains_cdr_length(self):
        chain_domain = ChainDomains(path=self.ab_file)
        self.assertEqual(chain_domain.cdr_lengths()[0, 0], 7)

    def test_ChainDomains_cdr_sequence(self):
        chain_domain = ChainDomains(path=self.ab_file)
        self.assertEqual(chain_domain.cdr_sequences()['Seq1']['CDR3'], 'GLRYTRAGMIWG')

    def test_ChainDomains_framework_length(self):
        chain_domain = ChainDomains(path=self.ab_file)
        self.assertEqual(chain_domain.framework_length()[0, 0], 25)

    def test_ChainDomains_framework_sequence(self):
        chain_domain = ChainDomains(path=self.ab_file)
        self.assertEqual(chain_domain.framework_sequences()['Seq1']['FR4'], 'WGQGTLVTVSS')