def test_FabCollection_input4(self): light_chain_collection = ChainCollection(path='./tests/Data/chain_collection_light_2_sequences.json') heavy_chain_collection = ChainCollection(path='./tests/Data/chain_collection_heavy_2_sequences.json') fab_collection = FabCollection(light_chains=light_chain_collection, heavy_chains=heavy_chain_collection, names=['Fab1', 'Fab2']) self.assertEqual(fab_collection.numbering_table()['Light']['CDR3']['L89'].loc['Fab1'], 'Q')
def test_ChainCollection_Hmatrix_calculation(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_fasta_test.fasta') antibody_collection_1.load(show_progressbar=False, verbose=False) # if this fails it means that abysis has been updated self.assertEqual(antibody_collection_1.hydrophobicity_matrix().shape, (1, 158))
def setUpClass(cls): cls.ab_collection_1 = ChainCollection.load_from_file(path='./tests/Data/sequence_alignment_seq_1.json', show_progressbar=False, verbose=False) cls.ab_collection_2 = ChainCollection.load_from_file(path='./tests/Data/sequence_alignment_seq_2.json', show_progressbar=False, verbose=False) cls.seq2_aligned = read_sequence_from_file('./tests/Data/BLOSUM62_aligned_sequence')
def test_ChainCollection_cdr_regions_part3_cdr(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) self.assertCountEqual( antibody_collection_1.ab_region_index() [self.antibody_collection_1_name]['CDR'], ['CDR1', 'CDR2', 'CDR3'])
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_total_charge(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) self.assertAlmostEqual( antibody_collection_1.total_charge[ self.antibody_collection_1_name], 1.3278508)
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_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_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_molecular_weight_monoisotopic(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) self.assertAlmostEqual( antibody_collection_1.molecular_weights(monoisotopic=True)[0], 20042.1121)
def test_ChainCollection_ec_reduced(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) self.assertAlmostEqual( antibody_collection_1.extinction_coefficients(reduced=True)[0], 52160.0)
def test_ChainCollection_append_1(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) antibody_collection_2 = ChainCollection( path='./tests/Data/chain_collection_2_heavy.json') antibody_collection_2.load(show_progressbar=False, verbose=False) antibody_collection_1.append(antibody_collection_2) self.assertEqual(antibody_collection_1.n_ab, 2)
def test_FabCollection_json1(self): light_chain_collection = ChainCollection.load_from_file( path='./tests/Data/chain_collection_light_2_sequences.json') heavy_chain_collection = ChainCollection.load_from_file( path='./tests/Data/chain_collection_heavy_2_sequences.json') fab_collection = FabCollection(light_chains=light_chain_collection, heavy_chains=heavy_chain_collection, names=['Fab1', 'Fab2']) fab_collection.save(file_format='json', path='./tests/fab_1') self.assertTrue(os.path.isfile('./tests/fab_1.json'))
def setUpClass(cls): cls.light_chain_collection = ChainCollection( path='./tests/Data/chain_collection_1_light.json') cls.heavy_chain_collection = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') cls.light_chain_collection.load(verbose=False, show_progressbar=False) cls.heavy_chain_collection.load(verbose=False, show_progressbar=False) cls.heavy_chain = cls.heavy_chain_collection[0] cls.light_chain = cls.light_chain_collection[0]
def test_FabCollection_json2(self): fab_collection = FabCollection.load_from_file( path='./tests/fab_1.json', show_progressbar=False, verbose=False) light_chain_collection = ChainCollection.load_from_file( path='./tests/Data/chain_collection_light_2_sequences.json') heavy_chain_collection = ChainCollection.load_from_file( path='./tests/Data/chain_collection_heavy_2_sequences.json') self.assertEqual(fab_collection._light_chains.sequences[0], light_chain_collection.sequences[0]) self.assertEqual(fab_collection._heavy_chains.sequences[0], heavy_chain_collection.sequences[0])
def test_ChainCollection_json(self): antibody_collection_1 = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') antibody_collection_1.load(show_progressbar=False, verbose=False) antibody_collection_1.save(file_format='json', file_path='./tests', file_name='SaveTest') antibody_collection_2 = ChainCollection(path='./tests/SaveTest.json') antibody_collection_2.load() self.assertEqual(antibody_collection_1.sequences[0], antibody_collection_2.sequences[0])
def setUpClass(cls): cls.test_sequence = read_sequence_from_single_sequence_fasta( './tests/Data/chain_collection_fasta_test.fasta') cls.heavy_chain_collection_object = ChainCollection( path='./tests/Data/chain_collection_1_heavy.json') cls.heavy_chain_collection_object.load() cls.heavy_chain_object = cls.heavy_chain_collection_object[0] cls.light_chain_collection_object = ChainCollection( path='./tests/Data/chain_collection_1_light.json') cls.light_chain_collection_object.load() cls.light_chain_object = cls.light_chain_collection_object[0]
def test_ChainCollection_append_2(self): antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) antibody_collection_2 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_2_heavy.json', show_progressbar=False, verbose=False) antibody_collection_1.append(antibody_collection_2) self.assertEqual(antibody_collection_1.hydrophobicity_matrix().shape, (2, 158))
def test_ChainCollection_add_exception_2(self): # check if adding two ChainCollection objects with one sequence each # results in a ChainCollection object with two sequences antibody_chothia = ChainCollection( path='./tests/Data/chain_collection_fasta_test.fasta', numbering_scheme='chothia') antibody_chothia.load(show_progressbar=False, verbose=False) antibody_kabat = Chain(sequence=read_sequence( './tests/Data/chain_collection_fasta_test.fasta'), numbering_scheme='kabat') antibody_kabat.load() self.assertRaises(ValueError, operator.add, antibody_chothia, antibody_kabat)
def test_ChainCollection_add(self): # check if adding two ChainCollection objects with one sequence each # results in a ChainCollection object with two sequences antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) antibody_collection_2 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_2_heavy.json', show_progressbar=False, verbose=False) antibody_collection_3 = antibody_collection_1 + antibody_collection_2 self.assertEqual(antibody_collection_3.n_ab, 2)
def test_ChainCollection_fasta(self): antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) antibody_collection_1.save(file_format='fasta', path='./tests/SaveTest') antibody_collection_2 = ChainCollection.load_from_file( path='./tests/SaveTest.fasta', show_progressbar=False, verbose=False) self.assertEqual(antibody_collection_1.sequences[0], antibody_collection_2.sequences[0])
def setUpClass(cls): cls.test_sequence = read_sequence( './tests/Data/chain_collection_fasta_test.fasta') cls.heavy_chain_collection_object = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) cls.heavy_chain_object = cls.heavy_chain_collection_object[0] cls.light_chain_collection_object = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_light.json', show_progressbar=False, verbose=False) cls.light_chain_object = cls.light_chain_collection_object[0]
def setUpClass(cls): cls.light_chain_collection = ChainCollection(path='./tests/Data/chain_collection_light_2_sequences.json') cls.heavy_chain_collection = ChainCollection(path='./tests/Data/chain_collection_heavy_2_sequences.json') cls.light_chain_collection.load(verbose=False, show_progressbar=False) cls.heavy_chain_collection.load(verbose=False, show_progressbar=False) cls.heavy_chain_1 = cls.heavy_chain_collection[0] cls.light_chain_1 = cls.light_chain_collection[0] cls.heavy_chain_2 = cls.heavy_chain_collection[1] cls.light_chain_2 = cls.light_chain_collection[1] cls.fab_1 = Fab(heavy_chain=cls.heavy_chain_1, light_chain=cls.light_chain_1, name='Fab1') cls.fab_2 = Fab(heavy_chain=cls.heavy_chain_2, light_chain=cls.light_chain_2, name='Fab2')
def test_ChainCollection_cdr_regions_part1(self): antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) self.assertCountEqual(antibody_collection_1.ab_region_index().keys(), [self.antibody_collection_1_name])
def test_Chain_abysis_parser_kabat(self): antibody = ChainCollection.load_from_file( path='./tests/Data/chain_collection_fasta_test.fasta', numbering_scheme='kabat', verbose=False, show_progressbar=False) self.assertEqual(antibody.numbering_table(as_array=True)[0][-1], '-')
def test_ChainCollection_charge(self): antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) self.assertAlmostEqual(antibody_collection_1.charge.sum(), 1.7497642167513607)
def test_ChainCollection_get_object_1(self): # check if get_object returns a Chain object antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) self.assertIsInstance(antibody_collection_1.get_object('test'), Chain)
def test_ChainCollection_get_object_2(self): # check if get_object returns a Chain object and keeps the information (i.e. name) antibody_collection_1 = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) self.assertEqual(antibody_collection_1.get_object('test').name, 'test')
def test_Chain_set_name(self): heavy_chain_object = ChainCollection.load_from_file( path='./tests/Data/chain_collection_1_heavy.json', show_progressbar=False, verbose=False) heavy_chain_object[0].set_name("new_name") self.assertEqual(heavy_chain_object[0].name, "new_name")
def test_Chain_abysis_parser(self): antibody = ChainCollection.load_from_file( path='./tests/Data/chain_collection_fasta_test.fasta', numbering_scheme='chothia', verbose=False, show_progressbar=False) self.assertEqual(antibody.chain, 'heavy')