def test_set_internal(self): tree = SegmentTree([1,2,3,4]) self.assertEquals(10, tree.sum(0, 3)) tree.set(1, 10) self.assertEquals(11, tree.sum(0, 1)) self.assertEquals(18, tree.sum(0, 3))
def test_set_leaf(self): tree = SegmentTree([1,2,3,4]) self.assertEquals(2, tree.sum(1, 1)) tree.set(1, 10) self.assertEquals(10, tree.sum(1, 1))