def test_can_add_new_label(self):
store = MutableGraphStore(self.store)
labels = store.node_labels(self.a)
self.assertEqual(labels, {"X"})
labels.add("Z")
self.assertEqual(store.node_labels(self.a), {"X", "Z"})
assert "Z" in set(store.node_labels())
def test_can_clear_labels(self):
store = MutableGraphStore(self.store)
labels = store.node_labels(self.b)
self.assertEqual(labels, {"X", "Y"})
labels.clear()
self.assertFalse(store.node_labels(self.b))
def test_can_discard_label(self):
store = MutableGraphStore(self.store)
labels = store.node_labels(self.a)
self.assertEqual(labels, {"X"})
labels.discard("Z")
self.assertEqual(store.node_labels(self.a), {"X"})
def test_can_remove_label(self):
store = MutableGraphStore(self.store)
labels = store.node_labels(self.a)
self.assertEqual(labels, {"X"})
labels.remove("X")
self.assertFalse(store.node_labels(self.a))
def test_can_add_existing_label(self):
store = MutableGraphStore(self.store)
labels = store.node_labels(self.a)
self.assertEqual(labels, {"X"})
labels.add("X")
self.assertEqual(store.node_labels(self.a), {"X"})
def test_should_create_copy_of_mutable_store(self):
store = MutableGraphStore(self.store)
self.assertEqual(store.node_count(), 4)
self.assertEqual(store.relationship_count(), 4)
self.assertEqual(store.node_labels(), {"X", "Y"})
self.assertEqual(store.relationship_types(), {"KNOWS"})
def test_should_create_empty_on_none(self):
store = MutableGraphStore()
self.assertEqual(store.node_count(), 0)
self.assertEqual(store.relationship_count(), 0)
self.assertFalse(store.node_labels())
self.assertFalse(store.relationship_types())