Exemplo n.º 1
0
    def test_leaf_internal_idx(self):
        sumTree = SumTree(10, [Item("A", 6), Item("A", 3), Item("A", 1), Item("A", 0), Item("A", 9), Item("A", 8),
                               Item("A", 5), Item("A", 8), Item("A", 7), Item("A", -1)])

        for idx in range(sumTree.get_num_leaves()):
            self.assertEqual(sumTree.tree[idx].idx, idx - sumTree.idx_shift)
Exemplo n.º 2
0
    def test_num_leaves(self):
        items = [Item("A", 6), Item("A", 3), Item("A", 1), Item("A", 0), Item("A", 9),
                 Item("A", 8), Item("A", 5), Item("A", 8), Item("A", 7), Item("A", -1)]
        sumTree = SumTree(10, items)

        self.assertEqual(sumTree.get_num_leaves(), len(items))