def test_setdefault_without_default_with_existing(self):
properties = PropertyDict({"name": "Alice", "age": 33})
value = properties.setdefault("age")
assert properties == {"name": "Alice", "age": 33}
assert value == 33
def test_setdefault_without_default_with_non_existent(self):
properties = PropertyDict({"name": "Alice"})
value = properties.setdefault("age")
assert properties == {"name": "Alice"}
assert value is None
def test_setter_with_none(self):
properties = PropertyDict({"name": "Alice", "age": 33})
properties["age"] = None
assert properties == {"name": "Alice"}
def test_setter_with_none_for_non_existent(self):
properties = PropertyDict({"name": "Alice"})
properties["age"] = None
assert properties == {"name": "Alice"}
def test_setter(self):
properties = PropertyDict({"name": "Alice"})
properties["age"] = 33
assert properties == {"name": "Alice", "age": 33}
def test_getter_with_none(self):
properties = PropertyDict({"name": "Alice"})
assert properties["age"] is None
def test_getter(self):
properties = PropertyDict({"name": "Alice"})
assert properties["name"] == "Alice"
def relationship_entry(cls, entry):
type_, nodes, properties = entry
return RelationshipEntry(type_, tuple(nodes), PropertyDict(properties))
def node_entry(self, key, entry):
def add_labels(*labels_):
for label in labels_:
self._nodes_by_label.setdefault(label, set()).add(key)
def remove_labels(*labels_):
for label in labels_:
try:
self._nodes_by_label[label].discard(key)
except KeyError:
pass
labels, properties = entry
return NodeEntry(ReactiveSet(labels, on_add=add_labels, on_remove=remove_labels), PropertyDict(properties))
