def test_should_get_node_labels(self):
store = FrozenGraphStore(self.store)
self.assertEqual(store.node_labels(), {"X", "Y"})
self.assertEqual(store.node_labels(self.a), {"X"})
self.assertEqual(store.node_labels(self.b), {"X", "Y"})
self.assertEqual(store.node_labels(self.c), {"X", "Y"})
self.assertEqual(store.node_labels(self.d), {"Y"})
self.assertIs(store.node_labels(object()), None)
def test_should_get_counts(self):
store = FrozenGraphStore(self.store)
self.assertEqual(store.node_count(), 4)
self.assertEqual(store.node_count("X"), 3)
self.assertEqual(store.relationship_count(), 6)
self.assertEqual(store.relationship_count("KNOWS"), 3)
self.assertEqual(store.node_labels(), {"X", "Y"})
self.assertEqual(store.relationship_types(),
{"LIKES", "KNOWS", "MARRIED_TO"})
def test_should_create_copy_of_mutable_store(self):
store = FrozenGraphStore(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 = FrozenGraphStore()
self.assertEqual(store.node_count(), 0)
self.assertEqual(store.relationship_count(), 0)
self.assertFalse(store.node_labels())
self.assertFalse(store.relationship_types())