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})