def test_ChainCollection_numbering_table_fr_region(self):
antibody_collection_1 = ChainCollection(
path='./tests/Data/chain_collection_1_heavy.json')
antibody_collection_1.load(show_progressbar=False, verbose=False)
self.assertEqual(
antibody_collection_1.numbering_table(
region='FR1').loc['test'].values[0], 'Q')
def test_ChainCollection_numbering_table_region_np(self):
antibody_collection_1 = ChainCollection(
path='./tests/Data/chain_collection_1_heavy.json')
antibody_collection_1.load(show_progressbar=False, verbose=False)
self.assertEqual(
antibody_collection_1.numbering_table(as_array=True,
region='CDR1')[0][-1], 'Y')
def test_ChainCollection_numbering_tableDataFrame(self):
antibody_collection_1 = ChainCollection(
path='./tests/Data/chain_collection_1_heavy.json')
antibody_collection_1.load(show_progressbar=False, verbose=False)
self.assertEqual(
antibody_collection_1.numbering_table(
as_array=False)['CDR1']['H32'].values[0], 'Y')
def test_ChainCollection_numbering_table_shape_pd(self):
antibody_collection_1 = ChainCollection(
path='./tests/Data/chain_collection_1_heavy.json')
antibody_collection_1.load(show_progressbar=False, verbose=False)
self.assertEqual(
antibody_collection_1.numbering_table(as_array=False).shape,
(1, 158))
def test_Chain_abysis_parser_kabat(self):
antibody = ChainCollection(
path='./tests/Data/chain_collection_fasta_test.fasta',
numbering_scheme='kabat')
antibody.load(verbose=False, show_progressbar=False)
self.assertEqual(antibody.numbering_table(as_array=True)[0][-1], '-')
