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