import phylotree as pt
from phylodm import PhyloDM
import sys

TREE = "test_trees/bench_trees/5000.test.nwk"

if __name__ == "__main__":
    if sys.argv[1] == "pdm":
        for _ in range(10):
            tree = PhyloDM.load_from_newick_path(TREE)
            dm = tree.dm()
            taxa = tree.taxa()
        print("Done PDM")
    else:
        for i in range(10):
            pdm = pt.DistanceMatrix.from_newick(TREE)
            dm = pdm.distances
            taxa = pdm.taxa
        print("Done Phylotree")