def test_add_from(self):
taxa = [
('A1', 'family a, subgroup 1'),
('A2', 'family a, subgroup 2'),
('B1a', 'family b, subgroup 1'),
('B1b', 'family b, subgroup 1'),
('B2', 'family b, subgroup 2'),
]
t = TreeMaker()
t.add_from(taxa)
assert str(t.tree) == "((A1,A2),((B1a,B1b),B2))"
def test_add_from(self):
taxa = [
('A1', 'family a, subgroup 1'),
('A2', 'family a, subgroup 2'),
('B1a', 'family b, subgroup 1'),
('B1b', 'family b, subgroup 1'),
('B2', 'family b, subgroup 2'),
]
t = TreeMaker()
t.add_from(taxa)
assert str(t.tree) == "((A1,A2),((B1a,B1b),B2))"
#!/usr/bin/env python
# coding=utf-8
import os
try:
from treemaker import TreeMaker
except ImportError:
raise ImportError("Please install treemaker")
from tools import get_table, safe_name
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser(
description="Constructs a tree from the glottolog data file")
parser.add_argument("glottfile", help="Glottolog data file")
parser.add_argument("treefile", help="Treefile to save as")
args = parser.parse_args()
if not os.path.isfile(args.glottfile):
raise IOError("File %s not found" % args.glottfile)
t = TreeMaker()
t.add_from((safe_name(o.Language), o.Classification)
for o in get_table(args.glottfile))
t.write_to_file(args.treefile)