def test_read_headerless(self):
# This example has blank headers.
f = StringIO(example3)
seqs = fasta_io.read(f)
self.assertEqual(len(seqs), 4)
# print seqs
fout = StringIO()
fasta_io.write(fout, seqs)
def test_write_comments(self):
f = StringIO(example_with_optional_comments)
seqs = fasta_io.read(f)
fout = StringIO()
fasta_io.write(fout, seqs)
fout.seek(0)
seqs2 = fasta_io.read(fout)
self.assertEqual(seqs, seqs2)
self.assertEqual(seqs[1].description, seqs2[1].description)
def test_write_seq(self):
f = StringIO(fasta_io.example)
seqs = fasta_io.read(f)
fout = StringIO()
fasta_io.write(fout, seqs)
fout.seek(0)
seqs2 = fasta_io.read(fout)
self.assertEqual(seqs, seqs2)
def test_write_with_header(self):
f = StringIO(fasta_io.example)
seqs = fasta_io.read(f)
seqs.description = "A description\nMore description"
fout = StringIO()
fasta_io.write(fout, seqs)