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