def test_sort_4(self):
sequence_file = SequenceFile("test")
for seq in [Sequence("foo", "AAAAA"), Sequence("bar", "DDDDD"), Sequence("doe", "CCCCC")]:
sequence_file.add(seq)
sequence_file_sorted = sequence_file.sort("seq", reverse=True, inplace=True)
self.assertEqual(["bar", "doe", "foo"], [s.id for s in sequence_file_sorted])
self.assertEqual(["DDDDD", "CCCCC", "AAAAA"], [s.seq for s in sequence_file_sorted])
self.assertEqual(sequence_file, sequence_file_sorted)
def test_sort_4(self):
sequence_file = SequenceFile('test')
for seq in [
Sequence('foo', 'AAAAA'),
Sequence('bar', 'DDDDD'),
Sequence('doe', 'CCCCC')
]:
sequence_file.add(seq)
sequence_file_sorted = sequence_file.sort('seq',
reverse=True,
inplace=True)
self.assertEqual(['bar', 'doe', 'foo'],
[s.id for s in sequence_file_sorted])
self.assertEqual(['DDDDD', 'CCCCC', 'AAAAA'],
[s.seq for s in sequence_file_sorted])
self.assertEqual(sequence_file, sequence_file_sorted)
def test_sort_2(self):
sequence_file = SequenceFile('test')
for seq in [
Sequence('foo', 'AAAAA'),
Sequence('bar', 'BBBBB'),
Sequence('doe', 'CCCCC')
]:
sequence_file.add(seq)
sequence_file_sorted = sequence_file.sort('id',
reverse=True,
inplace=False)
self.assertEqual(['foo', 'doe', 'bar'],
[s.id for s in sequence_file_sorted])
self.assertEqual(['AAAAA', 'CCCCC', 'BBBBB'],
[s.seq for s in sequence_file_sorted])
self.assertNotEqual(sequence_file, sequence_file_sorted)