def test_use_dict():
d = {1: 2}
with use(d) as ctx:
assert not ctx.is_changed(d)
d[3] = 4
assert d == {1: 2, 3: 4}
assert ctx.is_changed(d)
assert d == {1: 2}
def test_use_list():
ls = [1, 2, 3]
with use(ls) as ctx:
assert not ctx.is_changed(ls)
ls.append(5)
assert ls == [1, 2, 3, 5]
assert ctx.is_changed(ls)
assert ls == [1, 2, 3]
def test_use_set():
s = {1, 2}
with use(s) as ctx:
assert not ctx.is_changed(s)
s.add(15)
assert s == {1, 2, 15}
assert ctx.is_changed(s)
assert s == {1, 2}
def test_refs_global():
global g_a
with use(refs=['g_a']) as ctx:
assert not ctx.is_ref_changed('g_a')
g_a = 16
assert g_a == 16
assert ctx.is_ref_changed('g_a')
assert g_a == 15
def internal():
nonlocal a
with use(refs=['a']) as ctx:
assert not ctx.is_ref_changed('a')
a = 16
assert a == 16
assert ctx.is_ref_changed('a')
assert a == 15
def test_refs():
a = 15
with use(refs=['a']) as ctx:
assert not ctx.is_ref_changed('a')
a = 16
assert a == 16
assert ctx.is_ref_changed('a')
assert a == 15
def test_with_dict():
class A:
pass
a = A()
a.a = 1
with use(a) as ctx:
assert not ctx.is_changed(a)
a.a = 2
assert a.a == 2
assert ctx.is_changed(a)
assert a.a == 1
def test_user_str():
class A(str):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.value = 2
a = A()
with use(a) as ctx:
assert not ctx.is_changed(a)
a.value = 3
assert a.value == 3
assert ctx.is_changed(a)
assert a.value == 2
def test_dataclass():
from dataclasses import dataclass
@dataclass
class A:
f1: str
f2: str
f3: str
a = A('1', '2', '3')
with use(a) as ctx:
assert not ctx.is_changed(a)
a.f1 = '6'
assert a == A('6', '2', '3')
assert ctx.is_changed(a)
assert a == A('1', '2', '3')
def test_with_slots():
class A:
__slots__ = ('a', 'b')
a = A()
a.a = 1
with use(a) as ctx:
assert not ctx.is_changed(a)
a.a = 2
a.b = 3
assert a.a == 2
assert a.b == 3
assert ctx.is_changed(a)
assert a.a == 1
assert not hasattr(a, 'b')
def test_state():
class A:
def get_state(self):
return vars(self).copy()
def from_state(self, state):
vars(self).clear()
vars(self).update(state)
a = A()
a.a = 1
with use(a) as ctx:
assert not ctx.is_changed(a)
a.a = 2
assert a.a == 2
assert ctx.is_changed(a)
assert a.a == 1
def test_user_str_with_slots():
class A(str):
__slots__ = 'value'
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.value = 2
assert use_context._get_slots_attrs(A) == ['value']
a = A()
with use(a) as ctx:
assert not ctx.is_changed(a)
a.value = 3
assert a.value == 3
assert ctx.is_changed(a)
assert a.value == 2