def test_known_mappings(): with open('TestData/LocatorRes.tsv') as handle: cor_res = list(csv.DictReader(handle, delimiter='\t')) with open('TestData/testSeqs.fasta') as handle: test_seqs = list(fasta_reader(handle)) for row, crow in zip(HIVTransTool.process_seqs(test_seqs, extract_regions=True), cor_res): for f in crow.keys(): if row[f] is None: row[f] = '' yield eq_, str(row[f]), crow[f], f