def test_attrset_unset_error() -> None:
obj = SimpleNamespace(foo=42)
with pytest.raises(RuntimeError, match="Catch this!"):
with attrset(obj, "bar", "quux"):
assert obj.bar == "quux"
raise RuntimeError("Catch this!")
assert not hasattr(obj, "bar")
def test_attrset_error() -> None:
obj = SimpleNamespace(foo=42)
with pytest.raises(RuntimeError, match="Catch this!"):
with attrset(obj, "foo", "bar"):
assert obj.foo == "bar"
raise RuntimeError("Catch this!")
assert obj.foo == 42
def test_attrset_not_copied_error() -> None:
x = object()
obj = SimpleNamespace(foo=x)
with pytest.raises(RuntimeError, match="Catch this!"):
with attrset(obj, "foo", "bar"):
assert obj.foo == "bar"
raise RuntimeError("Catch this!")
assert obj.foo is x
def test_attrset_unset_delled() -> None:
obj = SimpleNamespace(foo=42)
with attrset(obj, "bar", "quux"):
assert obj.bar == "quux"
del obj.bar
assert not hasattr(obj, "bar")
def test_attrset_unset_modified() -> None:
obj = SimpleNamespace(foo=42)
with attrset(obj, "bar", "quux"):
assert obj.bar == "quux"
obj.bar = [3.14]
assert not hasattr(obj, "bar")
def test_attrset() -> None:
obj = SimpleNamespace(foo=42)
with attrset(obj, "foo", "bar"):
assert obj.foo == "bar"
assert obj.foo == 42
def test_attrset_modified() -> None:
obj = SimpleNamespace(foo=42)
with attrset(obj, "foo", "bar"):
assert obj.foo == "bar"
obj.foo = [3.14]
assert obj.foo == 42 # type: ignore[comparison-overlap]
def test_attrset_not_copied() -> None:
x = object()
obj = SimpleNamespace(foo=x)
with attrset(obj, "foo", "bar"):
assert obj.foo == "bar"
assert obj.foo is x