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)