def test_io(self):
# Very basic check that read/write public API is present and appears to
# be functioning. Roundtrip from memory -> disk -> memory and ensure
# results match.
fh = StringIO()
self.dm_3x3.write(fh)
fh.seek(0)
deserialized = DissimilarityMatrix.read(fh)
self.assertEqual(deserialized, self.dm_3x3)
self.assertTrue(type(deserialized) == DissimilarityMatrix)
#!/usr/bin/env python
from __future__ import print_function
from skbio.stats.distance import DissimilarityMatrix
from os import path
import sys
for distfile in sys.argv[1:]:
prefix = path.splitext(distfile)[0]
print(prefix)
dist = DissimilarityMatrix.read(distfile)
fig = dist.plot(title=prefix)
fig.tight_layout()
fig.savefig("{prefix}.pdf".format(prefix=prefix))
