#!/usr/bin/env python

import unittest

from weblogo.seq_io._nexus import Nexus

from . import data_stream


class test_nexus(unittest.TestCase):
    def test_create(self):
        n = Nexus()
        self.assertNotEqual(n, None)

    def test_parse_f0(self):
        f = data_stream("nexus/test_Nexus_input.nex")
        n = Nexus(f)
        # self.output_basics(n)

        expected = [
            "t1",
            "t2 the name",
            "isn'that [a] strange name?",
            "one should be punished, for (that)!",
            "t5",
            "t6",
            "t7",
            "t8",
            "t9",
        ]
        taxa = n.taxlabels
        self.assertEqual(taxa, expected)
        f.close()

    def test_parse_protein(self):
        f = data_stream("nexus/protein.nex")
        Nexus(f)
        f.close()

    def test_parse_dna(self):
        f = data_stream("nexus/dna.nex")
        n = Nexus(f)

        taxa = n.taxlabels
        taxa.sort()
        self.assertEqual(len(taxa), 10)
        self.assertEqual(taxa[0], "Carp")
        self.assertEqual(taxa[-1], "Whale")
        f.close()

    def test_TreeTest1(self):
        """Test Tree module."""
        f = data_stream("nexus/test_Nexus_input.nex")
        n = Nexus(f)
        t3 = n.trees[2]
        n.trees[2]
        t3.root_with_outgroup(["t1", "t5"])

        # Return node_id of common ancestor if
        # taxon_list is monophyletic, -1 otherwise.
        self.assertEqual(t3.is_monophyletic(["t1", "t5"]), 13)

        t3.split(parent_id=t3.search_taxon("t9"))
        f.close()


if __name__ == "__main__":
    unittest.main()
