def assert_rebalanced(tree):
exp_tree = Tree(copy.deepcopy(tree.vos), tree.label_grid)
tree.init()
exp_tree.init()
np.testing.assert_equal(tree.tree, exp_tree.tree)
def test_build(vos, label_grid):
tree = Tree(vos, label_grid)
tree.init()
exp_tree = {
"d0": {
1: {0, 1},
2: {2},
3: {3}
},
"d1": {
"hello": {0, 2},
"world": {1, 3}
},
}
np.testing.assert_equal(tree.tree, exp_tree)