def test_should_get_nodes_with_a_label(self):
store = FrozenGraphStore(self.store)
self.assertEqual(set(store.nodes("X")), {self.a, self.b, self.c})
self.assertEqual(set(store.nodes("Y")), {self.b, self.c, self.d})
self.assertFalse(set(store.nodes("Z")))
def test_should_get_nodes_with_multiple_labels(self):
store = FrozenGraphStore(self.store)
self.assertEqual(set(store.nodes("X", "Y")), {self.b, self.c})
self.assertFalse(set(store.nodes("X", "Z")))
def test_should_get_nodes(self):
store = FrozenGraphStore(self.store)
self.assertEqual(set(store.nodes()), {self.a, self.b, self.c, self.d})
