def test_linkage_to_d3_4_observations(self):
Z = array([[ 1. , 3. , 0.45015331, 2. ], # arr[0], cluster4
[ 0. , 2. , 1.29504919, 2. ], # arr[1], cluster5
[ 4. , 5. , 1.55180264, 4. ]]) # arr[2], cluster6
expected = {
"name": "cluster6",
"children": [
{
"name": "cluster4",
"children": [
{"name": "cluster1", "size": 10},
{"name": "cluster3", "size": 10},
]
},
{
"name": "cluster5",
"children": [
{"name": "cluster0", "size": 10},
{"name": "cluster2", "size": 10},
],
},
]
}
# n = len(Z)+1
# d3_dict = _do_linkage_to_d3(n, len(Z)+n-1, Z)
d3_dict = linkage_to_d3(Z)
self.assertDictEqual(expected, d3_dict)
def test_linkage_to_d3_2_observations(self):
Z = array([[ 0. , 1. , 0.45015331, 2. ]]) # arr[0], cluster2
expected = {
"name": "cluster2",
"children": [
{
"name": "cluster0",
"size": 10
},
{
"name": "cluster1",
"size": 10
}
]
}
d3_dict = linkage_to_d3(Z)
self.assertDictEqual(expected, d3_dict)
def test_linkage_to_d3_3_observations(self):
Z = array([[ 1. , 2. , 0.45015331, 2. ], # arr[0], cluster3
[ 0. , 3. , 1.29504919, 2. ], # arr[1], cluster4
])
expected = {
"name": "cluster4",
"children": [
{ "name": "cluster0", "size": 10 },
{
"name": "cluster3",
"children": [
{ "name": "cluster1", "size": 10 },
{ "name": "cluster2", "size": 10 },
]
},
]
}
d3_dict = linkage_to_d3(Z)
self.assertDictEqual(expected, d3_dict)
def test_linkage_to_d3_0_observations(self):
Z = array([])
expected = {}
d3_dict = linkage_to_d3(Z)
self.assertDictEqual(expected, d3_dict)