1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
|
from __future__ import absolute_import
from __future__ import print_function
import unittest
from .. import ClusterTree
from .datasets import *
class Test_ClusterTree(unittest.TestCase):
""" Tests specific methods for trees linked to ArrayTables"""
def test_clustertree(self):
""" Tests tree-ArrayTable association """
t = ClusterTree("(((A,B),(C,(D,E))),(F,(G,H)));", text_array=expression)
# Now we can ask the expression profile of a single gene
node = t.get_common_ancestor("C", "D", "E")
self.assertEqual((t&"A").profile.tolist(), \
[-1.23, -0.81, 1.79, 0.78,-0.42,-0.69, 0.58])
print(node.profile)
print(node.deviation)
print(node.silhouette)
print(node.intracluster_dist)
print(node.intercluster_dist)
c1 = t.get_common_ancestor("A", "B")
c2 = t.get_common_ancestor("C", "D", "E")
c3 = t.get_common_ancestor("F", "G", "H")
print(t.get_dunn([c1, c2, c3]))
|