def test_setting():
class Obj():
pass
obj = Obj()
obj.myattr = 'old'
with setting(obj, 'myattr', 'mid'):
assert obj.myattr == 'mid'
with setting(obj, 'myattr', 'new'):
assert obj.myattr == 'new'
assert obj.myattr == 'mid'
assert obj.myattr == 'old'
with setting(obj, 'otherattr', 'mid'):
assert obj.otherattr == 'mid'
with setting(obj, 'otherattr', 'new'):
assert obj.otherattr == 'new'
with setting(obj, 'anotherattr', 'value'):
assert obj.anotherattr == 'value'
assert not hasattr(obj, 'anotherattr')
assert obj.otherattr == 'mid'
assert not hasattr(obj, 'otherattr')
def test_collection():
with setting(config, 'GC_NBYTES_THRESHOLD', 8000):
memory.garbage_collect()
counter = 0
for i in range(10):
data = np.arange(100)
counter += data.nbytes
op = DiagonalOperator(data)
op.delete()
if i < 9:
assert memory._gc_nbytes_counter == counter
assert memory._gc_nbytes_counter == 0
def test_setting():
class Obj:
pass
obj = Obj()
obj.myattr = "old"
with setting(obj, "myattr", "mid"):
assert obj.myattr == "mid"
with setting(obj, "myattr", "new"):
assert obj.myattr == "new"
assert obj.myattr == "mid"
assert obj.myattr == "old"
with setting(obj, "otherattr", "mid"):
assert obj.otherattr == "mid"
with setting(obj, "otherattr", "new"):
assert obj.otherattr == "new"
with setting(obj, "anotherattr", "value"):
assert obj.anotherattr == "value"
assert not hasattr(obj, "anotherattr")
assert obj.otherattr == "mid"
assert not hasattr(obj, "otherattr")
