def test_roundtrip(self):
for tree, newicks in self.trees_newick_lists:
newick = newicks[0]
fh = StringIO(newick)
tree = _newick_to_tree_node(fh)
fh2 = StringIO()
_tree_node_to_newick(tree, fh2)
fh2.seek(0)
tree2 = _newick_to_tree_node(fh2)
self.assertEqual(newick, fh2.getvalue())
self._assert_equal(tree, tree2)
fh.close()
fh2.close()
def test_roundtrip_read_write(self):
for reader_fn, writer_fn, fhs in (
(_lsmat_to_dissimilarity_matrix, _dissimilarity_matrix_to_lsmat, self.dissim_fhs),
(_lsmat_to_distance_matrix, _distance_matrix_to_lsmat, self.dist_fhs),
):
for fh in fhs:
# Read.
fh.seek(0)
lsmat1 = reader_fn(fh)
# Write.
out_fh = StringIO()
writer_fn(lsmat1, out_fh)
out_fh.seek(0)
# Read.
lsmat2 = reader_fn(out_fh)
out_fh.close()
self.assertEqual(lsmat1, lsmat2)
def test_roundtrip_read_write(self):
for reader_fn, writer_fn, fhs in ((_lsmat_to_dissimilarity_matrix,
_dissimilarity_matrix_to_lsmat,
self.dissim_fhs),
(_lsmat_to_distance_matrix,
_distance_matrix_to_lsmat,
self.dist_fhs)):
for fh in fhs:
# Read.
fh.seek(0)
lsmat1 = reader_fn(fh)
# Write.
out_fh = StringIO()
writer_fn(lsmat1, out_fh)
out_fh.seek(0)
# Read.
lsmat2 = reader_fn(out_fh)
out_fh.close()
self.assertEqual(lsmat1, lsmat2)
