Ejemplos de SumTree.get_num_leaves en Python

Ejemplo n.º 1

Archivo: PER_Test.py Proyecto: Grey-Poupon/RL-Experiments

    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)

Ejemplo n.º 2

Archivo: PER_Test.py Proyecto: Grey-Poupon/RL-Experiments

    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))