def test_truncate_seqs(self): """ Truncates seqs, scores correctly """ base_pos = 5 fasta_seqs = {"seq1": "GAAATCAAGAATAC", "seq2": "ATAAACAAGAT"} qual_scores = { "seq1": array(map(str, [20, 10, 15, 25, 24, 25, 27])), "seq2": array(map(str, [22, 21, 15, 12, 22, 25, 27, 28])), } expected_fasta = {"seq1": "GAAAT", "seq2": "ATAAA"} expected_qual = {"seq1": map(str, array([20, 10, 15, 25, 24])), "seq2": map(str, array([22, 21, 15, 12, 22]))} actual_fasta_seqs, actual_qual_scores = truncate_seqs(fasta_seqs, qual_scores, base_pos) self.assertDictEqual(actual_fasta_seqs, expected_fasta) self.assertItemsEqual(expected_qual, actual_qual_scores)
def test_truncate_seqs(self): """ Truncates seqs, scores correctly """ base_pos = 5 fasta_seqs = {'seq1':'GAAATCAAGAATAC', 'seq2':'ATAAACAAGAT'} qual_scores = {'seq1':array(map(str,[20, 10, 15, 25, 24, 25, 27])), 'seq2':array(map(str,[22, 21, 15, 12, 22, 25, 27, 28]))} expected_fasta = {'seq1':'GAAAT', 'seq2':'ATAAA'} expected_qual = {'seq1':map(str,array([20, 10, 15, 25, 24])), 'seq2':map(str,array([22, 21, 15, 12, 22]))} actual_fasta_seqs, actual_qual_scores =\ truncate_seqs(fasta_seqs, qual_scores, base_pos) self.assertEqual(actual_fasta_seqs, expected_fasta) self.assertEqual(actual_qual_scores, expected_qual)
def test_truncate_seqs(self): """ Truncates seqs, scores correctly """ base_pos = 5 fasta_seqs = {'seq1': 'GAAATCAAGAATAC', 'seq2': 'ATAAACAAGAT'} qual_scores = {'seq1': array(map(str, [20, 10, 15, 25, 24, 25, 27])), 'seq2': array(map(str, [22, 21, 15, 12, 22, 25, 27, 28]))} expected_fasta = {'seq1': 'GAAAT', 'seq2': 'ATAAA'} expected_qual = {'seq1': map(str, array([20, 10, 15, 25, 24])), 'seq2': map(str, array([22, 21, 15, 12, 22]))} actual_fasta_seqs, actual_qual_scores =\ truncate_seqs(fasta_seqs, qual_scores, base_pos) self.assertDictEqual(actual_fasta_seqs, expected_fasta) self.assertItemsEqual(expected_qual, actual_qual_scores)