def test_phenotype_IO(self):
'''Test basic functionalities of phenotype IO methods'''
p1 = phenotype.read(SMALL_JSON_PLATE, 'pm-json')
p2 = next(phenotype.parse(SMALL_CSV_PLATES, 'pm-csv'))
handle = StringIO()
c = phenotype.write([p1, p2], handle, 'pm-json')
self.assertEqual(c, 2)
handle.flush()
handle.seek(0)
# Now ready to read back from the handle...
try:
records = list(phenotype.parse(handle, 'pm-json'))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError("%s\n\n%s\n\n%s"
% (str(e), repr(handle.read()), repr(records)))
self.assertEqual(p1, records[0])
handle.close()
handle = StringIO()
self.assertRaises(TypeError, phenotype.write, p1, handle, 1)
self.assertRaises(ValueError, phenotype.write, p1, handle, 'PM-JSON')
self.assertRaises(ValueError, phenotype.write, p1, handle, 'pm-csv')
handle.close()
def test_phenotype_IO(self):
"""Test basic functionalities of phenotype IO methods."""
p1 = phenotype.read(SMALL_JSON_PLATE, "pm-json")
p2 = next(phenotype.parse(SMALL_CSV_PLATES, "pm-csv"))
handle = StringIO()
c = phenotype.write([p1, p2], handle, "pm-json")
self.assertEqual(c, 2)
handle.flush()
handle.seek(0)
# Now ready to read back from the handle...
try:
records = list(phenotype.parse(handle, "pm-json"))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError("%s\n\n%s\n\n%s" %
(str(e), repr(handle.read()), repr(records)))
self.assertEqual(p1, records[0])
handle.close()
handle = StringIO()
self.assertRaises(TypeError, phenotype.write, p1, handle, 1)
self.assertRaises(ValueError, phenotype.write, p1, handle, "PM-JSON")
self.assertRaises(ValueError, phenotype.write, p1, handle, "pm-csv")
handle.close()
def check_simple_write_read(alignments, indent=" "):
# print(indent+"Checking we can write and then read back these alignments")
for format in test_write_read_align_with_seq_count:
records_per_alignment = len(alignments[0])
for a in alignments:
if records_per_alignment != len(a):
records_per_alignment = None
# Can we expect this format to work?
if not records_per_alignment \
and format not in test_write_read_alignment_formats:
continue
print(indent + "Checking can write/read as '%s' format" % format)
# Going to write to a handle...
handle = StringIO()
try:
c = AlignIO.write(alignments, handle=handle, format=format)
assert c == len(alignments)
except ValueError as e:
# This is often expected to happen, for example when we try and
# write sequences of different lengths to an alignment file.
print(indent + "Failed: %s" % str(e))
# Carry on to the next format:
continue
# First, try with the seq_count
if records_per_alignment:
handle.flush()
handle.seek(0)
try:
alignments2 = list(
AlignIO.parse(handle=handle,
format=format,
seq_count=records_per_alignment))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError(
"%s\n\n%s\n\n%s" %
(str(e), repr(handle.read()), repr(alignments2)))
simple_alignment_comparison(alignments, alignments2, format)
if format in test_write_read_alignment_formats:
# Don't need the seq_count
handle.flush()
handle.seek(0)
try:
alignments2 = list(AlignIO.parse(handle=handle, format=format))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError(
"%s\n\n%s\n\n%s" %
(str(e), repr(handle.read()), repr(alignments2)))
simple_alignment_comparison(alignments, alignments2, format)
if len(alignments) > 1:
# Try writing just one Alignment (not a list)
handle = StringIO()
AlignIO.write(alignments[0:1], handle, format)
assert handle.getvalue() == alignments[0].format(format)
def check_simple_write_read(alignments, indent=" "):
# print(indent+"Checking we can write and then read back these alignments")
for format in test_write_read_align_with_seq_count:
records_per_alignment = len(alignments[0])
for a in alignments:
if records_per_alignment != len(a):
records_per_alignment = None
# Can we expect this format to work?
if not records_per_alignment \
and format not in test_write_read_alignment_formats:
continue
print(indent + "Checking can write/read as '%s' format" % format)
# Going to write to a handle...
handle = StringIO()
try:
c = AlignIO.write(alignments, handle=handle, format=format)
assert c == len(alignments)
except ValueError as e:
# This is often expected to happen, for example when we try and
# write sequences of different lengths to an alignment file.
print(indent + "Failed: %s" % str(e))
# Carry on to the next format:
continue
# First, try with the seq_count
if records_per_alignment:
handle.flush()
handle.seek(0)
try:
alignments2 = list(AlignIO.parse(handle=handle, format=format,
seq_count=records_per_alignment))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError("%s\n\n%s\n\n%s"
% (str(e), repr(handle.read()), repr(alignments2)))
simple_alignment_comparison(alignments, alignments2, format)
if format in test_write_read_alignment_formats:
# Don't need the seq_count
handle.flush()
handle.seek(0)
try:
alignments2 = list(AlignIO.parse(handle=handle, format=format))
except ValueError as e:
# This is BAD. We can't read our own output.
# I want to see the output when called from the test harness,
# run_tests.py (which can be funny about new lines on Windows)
handle.seek(0)
raise ValueError("%s\n\n%s\n\n%s"
% (str(e), repr(handle.read()), repr(alignments2)))
simple_alignment_comparison(alignments, alignments2, format)
if len(alignments) > 1:
# Try writing just one Alignment (not a list)
handle = StringIO()
SeqIO.write(alignments[0], handle, format)
assert handle.getvalue() == alignments[0].format(format)
