def test_RunBlast(self): blast = RunBlast(records=self.record_index, e_value=self.e_threshold, wdir=self.Blast_dir) blast.run_single() expected = dict({'lcl|AE014075.1_gene_1': set([]), 'lcl|AE014075.1_gene_2': set(['GO:0004803', 'GO:0006313']), 'lcl|AE014075.1_gene_3': set(['GO:0071470', 'GO:0016310', 'GO:0005886', 'GO:0009067', 'GO:0000023', 'GO:0016597', 'GO:0043085', 'GO:0016491', 'GO:0005737', 'GO:0050661', 'GO:0040007', 'GO:0005618', 'GO:0009570', 'GO:0005634', 'GO:0006520', 'GO:0019877', 'GO:0000166', 'GO:0016740', 'GO:0009097', 'GO:0009090', 'GO:0019252', 'GO:0019761', 'GO:0016301', 'GO:0008152', 'GO:0009088', 'GO:0055114', 'GO:0009507', 'GO:0008652', 'GO:0005829', 'GO:0006555', 'GO:0004412', 'GO:0005575', 'GO:0009089', 'GO:0005524', 'GO:0006164', 'GO:0006531', 'GO:0009086', 'GO:0004072', 'GO:0009082']), 'lcl|AE014075.1_gene_4': set(['GO:0005737', 'GO:0006566', 'GO:0000394', 'GO:0016310', 'GO:0009617', 'GO:0004413', 'GO:0000166', 'GO:0019344', 'GO:0009620', 'GO:0009088', 'GO:0009570', 'GO:0009086', 'GO:0005524', 'GO:0009507']), 'lcl|AE014075.1_gene_5': set(['GO:0005125', 'GO:0016311', 'GO:0046360', 'GO:0003674', 'GO:0030170', 'GO:0004795', 'GO:0005737', 'GO:0006566', 'GO:0005615', 'GO:0005634', 'GO:0006520', 'GO:0005524', 'GO:0008150', 'GO:0070905', 'GO:0008152', 'GO:0009071', 'GO:0008652', 'GO:0006897', 'GO:0005829', 'GO:0005575', 'GO:0009088', 'GO:0004765', 'GO:0016829']) }) for k, v in expected.items(): seq = blast.results[k] self.assertEqual(v, seq.combined_terms, "Error in %s" % k)
def test_RunBlast_subset(self): sub_record = dict({'lcl|AE014075.1_gene_4': self.record_index['lcl|AE014075.1_gene_4'], 'lcl|AE014075.1_gene_5': self.record_index['lcl|AE014075.1_gene_5'] }) blast = RunBlast(records=sub_record, e_value=1e-200, wdir=self.Blast_dir) blast.run_single() expected = dict({'lcl|AE014075.1_gene_4': set([]), # 'lcl|AE014075.1_gene_5':set(['GO:0004795', 'GO:0030170', 'GO:0009088']) 'lcl|AE014075.1_gene_5': set(['GO:0004795', 'GO:0009088']) }) for k, v in expected.items(): seq = blast.results[k] self.assertEqual(v, seq.combined_terms) # for k, v in blast.results.items(): print k, v.combined_terms