def test_discard_with_attvars(): py.test.skip("not implemented yet") h0 = Heap() v0 = h0.new_attvar() h1 = h0.branch() v1 = h1.new_attvar() h2 = h1.branch() v2 = h2.new_attvar() h2.add_trail_atts(v0, "m") v0.atts = {"m": 1} h2.add_trail_atts(v1, "n") v1.atts = {"n": 2} h3 = h2.branch() h3.add_trail_atts(v2, "a") v2.atts = {"a": 3} h = h2.discard(h3) assert h3.prev is h1 assert h3 is h assert h3.revert_upto(h0) assert v0.atts == {} assert v1.atts == {} assert v2.atts == {"a": 3}
def test_add_trail_atts(): hp = Heap() a = hp.new_attvar() assert a.created_after_choice_point is hp assert hp.trail_attrs is None ma = AttMap() ma.indexes = {"a": 0} a.value_list = [10] a.attmap = ma hp.add_trail_atts(a, "a") assert hp.trail_attrs is None hp2 = hp.branch() hp2.add_trail_atts(a, "a") assert hp2.trail_attrs == [(a, 0, 10)] a.add_attribute("a", 20) assert a.value_list == [20] hp2._revert() assert a.value_list == [10] hp3 = hp2.branch() hp3.add_trail_atts(a, "b") a.add_attribute("b", 30) assert a.value_list == [10, 30] assert a.attmap.indexes == {"a": 0, "b": 1} assert a.attmap is not ma hp3._revert() assert a.value_list == [10, None]
def test_heap_dont_trail_new_attvars(): h1 = Heap() v1 = h1.new_attvar() h1.add_trail_atts(v1, "m") v1.add_attribute("m", 1) h2 = h1.branch() v2 = h2.new_attvar() h2.add_trail_atts(v1, "m") v1.add_attribute("m", 2) h2.add_trail_atts(v2, "m") v2.add_attribute("m", 3) h3 = h2.revert_upto(h1) t1 = v1.get_attribute("m") assert t1[0] == 1 assert t1[1] == 0 t2 = v2.get_attribute("m") # wasn't undone, because v2 dies assert t2[0] == 3 assert t2[1] == 0 assert h3 is h2
def test_new_attvar(): h = Heap() v = h.new_attvar() assert isinstance(v, AttVar) assert v.created_after_choice_point is h