def test_calculate_freq_3(self):
sequence_file = SequenceFile('test')
for s in [
Sequence('foo', 'AAAAAAA'),
Sequence('bar', 'AAAAAAA'),
Sequence('cho', 'AAAAAAA')
]:
sequence_file.add(s)
calculated_freqs = [
round(i, 6) for i in sequence_file.calculate_freq()
]
self.assertEqual([1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], calculated_freqs)
def test_calculate_freq_1(self):
sequence_file = SequenceFile('test')
for s in [
Sequence('foo', 'AAAAAAA'),
Sequence('bar', 'A-AAAA-'),
Sequence('cho', '--AAA--')
]:
sequence_file.add(s)
calculated_freqs = [
round(i, 6) for i in sequence_file.calculate_freq()
]
self.assertEqual(
[0.666667, 0.333333, 1.0, 1.0, 1.0, 0.666667, 0.333333],
calculated_freqs)