def test_calculate_weights_5(self):
sequence_file = SequenceFile('test')
for s in [
Sequence('foo', 'AAAAAAA'),
Sequence('bar', 'AA-ABA-'),
Sequence('cho', 'B-BAA--'),
Sequence('baz', 'BBBBBBB')
]:
sequence_file.add(s)
weights = sequence_file.calculate_weights(identity=0.6)
self.assertEqual(weights, [1.0, 1.0, 1.0, 1.0])
def test_calculate_weights_1(self):
sequence_file = SequenceFile('test')
for s in [
Sequence('foo', 'AAAAAAA'),
Sequence('bar', 'AAAAAAA'),
Sequence('cho', 'AAAAAAA'),
Sequence('baz', 'AAAAAAA')
]:
sequence_file.add(s)
weights = sequence_file.calculate_weights(identity=0.7)
self.assertEqual(weights, [0.25, 0.25, 0.25, 0.25])
def test_calculate_weights_6(self):
sequence_file = SequenceFile('test')
for s in [
Sequence('foo', 'AAAAAAA'),
Sequence('bar', 'AA-ABA-'),
Sequence('cho', 'AAACBAA'),
Sequence('doo', 'B-BAA--'),
Sequence('miu', 'BBBBBBB'),
Sequence('nop', 'AAAAAAB')
]:
sequence_file.add(s)
weights = sequence_file.calculate_weights(identity=0.6)
self.assertEqual(weights,
[0.3333333333333333, 1.0, 0.5, 1.0, 1.0, 0.5])