def test_read_3(self):
msa = """#foo
#bar
>seq1
GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYF
>seq2
EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKA
>seq3
EVHKVQECKQSDIMMRDNLFEIVTTSRTFWKRRYFQLDENTIGYF
"""
f_name = create_tmp_f(content=msa)
parser = FastaParser()
with open(f_name, 'r') as f_in:
sequence_file = parser.read(f_in)
self.assertEqual(['foo', 'bar'], sequence_file.remark)
for i, sequence_entry in enumerate(sequence_file):
if i == 0:
self.assertEqual('seq1', sequence_entry.id)
self.assertEqual(
'GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYF',
sequence_entry.seq)
elif i == 1:
self.assertEqual('seq2', sequence_entry.id)
self.assertEqual(
'EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKA',
sequence_entry.seq)
elif i == 2:
self.assertEqual('seq3', sequence_entry.id)
self.assertEqual(
'EVHKVQECKQSDIMMRDNLFEIVTTSRTFWKRRYFQLDENTIGYF',
sequence_entry.seq)
os.unlink(f_name)
def test_read_3(self):
msa = """#foo
#bar
>seq1
GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYF
>seq2
EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKA
>seq3
EVHKVQECKQSDIMMRDNLFEIVTTSRTFWKRRYFQLDENTIGYF
"""
f_name = self.tempfile(content=msa)
parser = FastaParser()
with open(f_name, "r") as f_in:
sequence_file = parser.read(f_in)
self.assertEqual(["foo", "bar"], sequence_file.remark)
for i, sequence_entry in enumerate(sequence_file):
if i == 0:
self.assertEqual("seq1", sequence_entry.id)
self.assertEqual("GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYF", sequence_entry.seq)
elif i == 1:
self.assertEqual("seq2", sequence_entry.id)
self.assertEqual("EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKA", sequence_entry.seq)
elif i == 2:
self.assertEqual("seq3", sequence_entry.id)
self.assertEqual("EVHKVQECKQSDIMMRDNLFEIVTTSRTFWKRRYFQLDENTIGYF", sequence_entry.seq)
def test_write_2(self):
seq = [
"# Hello World",
">00FAF_A|<unknown description>",
"GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK",
]
f_name_in = self.tempfile(content="\n".join(seq))
f_name_out = self.tempfile()
parser = FastaParser()
with open(f_name_in, "r") as f_in, open(f_name_out, "w") as f_out:
sequence_file = parser.read(f_in)
parser.write(f_out, sequence_file)
with open(f_name_out, "r") as f_in:
output = f_in.read().splitlines()
self.assertEqual(seq, output)
def test_read_1(self):
seq = """>00FAF_A <unknown description>
GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK
EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKAVSGAIVAQRGPGRSA
SSEHP
"""
f_name = self.tempfile(content=seq)
parser = FastaParser()
with open(f_name, "r") as f_in:
sequence_file = parser.read(f_in)
sequence_entry = sequence_file.top_sequence
ref_id = "00FAF_A <unknown description>"
self.assertEqual(ref_id, sequence_entry.id)
ref_seq = "GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLKEVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKAVSGAIVAQRGPGRSASSEHP"
self.assertEqual(ref_seq, sequence_entry.seq)
def test_write_2(self):
seq = [
"# Hello World",
">00FAF_A|<unknown description>",
"GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK",
]
f_name_in = create_tmp_f(content='\n'.join(seq))
f_name_out = create_tmp_f()
parser = FastaParser()
with open(f_name_in, 'r') as f_in, open(f_name_out, 'w') as f_out:
sequence_file = parser.read(f_in)
parser.write(f_out, sequence_file)
with open(f_name_out, 'r') as f_in:
output = f_in.read().splitlines()
self.assertEqual(seq, output)
map(os.unlink, [f_name_in, f_name_out])
def test_write_1(self):
seq = [
">00FAF_A|<unknown description>",
"GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK",
"EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKAVSGAIVAQRGPGRSA",
"SSEHP",
]
f_name_in = self.tempfile(content="\n".join(seq))
f_name_out = self.tempfile()
parser = FastaParser()
with open(f_name_in, "r") as f_in, open(f_name_out, "w") as f_out:
sequence_file = parser.read(f_in)
parser.write(f_out, sequence_file)
with open(f_name_out, "r") as f_in:
output = f_in.read().splitlines()
self.assertEqual(seq, output)
def test_read_2(self):
seq = """# Hello World
>00FAF_A <unknown description>
GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK
"""
f_name = self.tempfile(content=seq)
parser = FastaParser()
with open(f_name, "r") as f_in:
sequence_file = parser.read(f_in)
sequence_entry = sequence_file.top_sequence
ref_f_remark = [" Hello World"]
self.assertEqual(ref_f_remark, sequence_file.remark)
ref_id = "00FAF_A <unknown description>"
self.assertEqual(ref_id, sequence_entry.id)
ref_seq = "GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK"
self.assertEqual(ref_seq, sequence_entry.seq)
def test_write_1(self):
seq = [
">00FAF_A|<unknown description>",
"GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYFKSELEKEPLRVIPLK",
"EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKAVSGAIVAQRGPGRSA",
"SSEHP",
"",
]
seq = os.linesep.join(seq)
f_name_in = create_tmp_f(content=seq)
f_name_out = create_tmp_f()
parser = FastaParser()
with open(f_name_in, 'r') as f_in, open(f_name_out, 'w') as f_out:
sequence_file = parser.read(f_in)
parser.write(f_out, sequence_file)
with open(f_name_out, 'r') as f_in:
output = "".join(f_in.readlines())
self.assertEqual(seq, output)
os.unlink(f_name_in)
os.unlink(f_name_out)
def test_write_3(self):
msa = [
"#foo",
"#bar",
">seq1",
"GSMFTPKPPQDSAVIKAGYCVKQGAVMKNWKRRYFQLDENTIGYF",
">seq2",
"EVHKVQECKQSDIMMRDNLFEIVTTSRTFYVQADSPEEMHSWIKA",
">seq3",
"EVHKVQECKQSDIMMRDNLFEIVTTSRTFWKRRYFQLDENTIGYF",
]
f_name_in = create_tmp_f(content='\n'.join(msa))
f_name_out = create_tmp_f()
parser = FastaParser()
with open(f_name_in, 'r') as f_in, open(f_name_out, 'w') as f_out:
sequence_file = parser.read(f_in)
parser.write(f_out, sequence_file)
with open(f_name_out, 'r') as f_in:
output = f_in.read().splitlines()
self.assertEqual(msa, output)
