def testRemove(self): tree = SampleTree(degree=5) for i in range(0,1000): tree.insert(str(i), i) for node in tree.nodes(): if node.weight == 100: tree.remove(node) break for node in tree.nodes(): if node.weight == 3: tree.remove(node) break assert tree.size() == 998 assert tree.weight() == (999 * 1000) / 2 - 103
def testNodes(self): tree = SampleTree(degree=2) for i in range(0,1000): tree.insert(str(i), i) assert tree.size() == 1000 assert tree.weight() == (999 * 1000) / 2 count = 0 weight = 0 for node in tree.nodes(): if node.data is not None: count += 1 weight += node.weight assert count == tree.size() assert weight == tree.weight()