Python Tree.init Examples

Example #1

File: test_tree.py Project: saharnazb/ParamTools

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)

Example #2

File: test_tree.py Project: saharnazb/ParamTools

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)