def test_should_get_node_degree(self):
store = FrozenGraphStore(self.store)
self.assertEqual(store.relationship_count(n_ids={self.a}), 4)
self.assertEqual(
store.relationship_count(r_type="LIKES", n_ids={self.a}), 2)
self.assertEqual(store.relationship_count(n_ids={self.b}), 4)
self.assertEqual(store.relationship_count(n_ids={self.c}), 3)
self.assertEqual(store.relationship_count(n_ids={self.d}), 1)
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())
