def iter_superclasses(class_): """Iterate over all the superclasses (and superclasses thereof, etc.) of given class. :param class_: Class to yield the superclasses of :return: Iterable of superclasses, super-superclasses, etc. of ``class_`` .. note:: In most cases, the result of :func:`iter_superclasses` is the same as ``class_.__mro__``, except when the method resolution order has been customized by the metaclass of ``class_``. """ ensure_class(class_) classes = set() def ascend(class_): superclasses = set(class_.__bases__) - classes classes.update(superclasses) return superclasses result = breadth_first(class_, ascend) next(result) # omit ``class_`` itself return result
def test_start__flat_graph(self): nodes = list(map(self._create_node, range(1, 5 + 1))) graph = self._create_node(0, nodes) bfs = __unit__.breadth_first(graph, self.CHILDREN_FUNC) for i, node in enumerate(bfs, 0): self.assertEquals(i, node.value)
def test_start__flat_graph(self): nodes = list(map(self._create_node, range(1, 5 + 1))) graph = self._create_node(0, nodes) bfs = __unit__.breadth_first(graph, self.CHILDREN_FUNC) for i, node in enumerate(bfs, 0): self.assertEquals(i, node.value)
def iter_superclasses(class_): """Iterate over all the superclasses (and superclasses thereof, etc.) of given class. :param class_: Class to yield the superclasses of :return: Iterable of superclasses, super-superclasses, etc. of ``class_`` .. note:: In most cases, the result of :func:`iter_superclasses` is the same as ``class_.__mro__``, except when the method resolution order has been customized by the metaclass of ``class_``. """ ensure_class(class_) classes = set() def ascend(class_): superclasses = set(class_.__bases__) - classes classes.update(superclasses) return superclasses result = breadth_first(class_, ascend) next(result) # omit ``class_`` itself return result
def iter_subclasses(class_): """Iterate over all the subclasses (and subclasses thereof, etc.) of given class. :param class_: Class to yield the subclasses of :return: Iterable of subclasses, sub-subclasses, etc. of ``class_`` """ ensure_class(class_) classes = set() def descend(class_): subclasses = set(class_.__subclasses__()) - classes classes.update(subclasses) return subclasses result = breadth_first(class_, descend) next(result) # omit ``class_`` itself return result
def iter_subclasses(class_): """Iterate over all the subclasses (and subclasses thereof, etc.) of given class. :param class_: Class to yield the subclasses of :return: Iterable of subclasses, sub-subclasses, etc. of ``class_`` """ ensure_class(class_) classes = set() def descend(class_): subclasses = set(class_.__subclasses__()) - classes classes.update(subclasses) return subclasses result = breadth_first(class_, descend) next(result) # omit ``class_`` itself return result
def test_start__none(self): bfs = __unit__.breadth_first(None, expand=functions.empty()) self.assertItemsEqual([None], bfs)
def test_expand__none(self): with self.assertRaises(TypeError): __unit__.breadth_first(self._create_node(), expand=None)
def test_expand__some_object(self): with self.assertRaises(TypeError): __unit__.breadth_first(self._create_node(), expand=object())
def test_start__path(self): graph = self._create_path(10) bfs = __unit__.breadth_first(graph, self.CHILDREN_FUNC) for i, node in enumerate(bfs, 0): self.assertEquals(i, node.value)
def test_start__none(self): bfs = __unit__.breadth_first(None, expand=functions.empty()) self.assertItemsEqual([None], bfs)
def test_start__some_object(self): obj = object() bfs = __unit__.breadth_first(obj, expand=functions.empty()) self.assertItemsEqual([obj], bfs)
def test_start__single_node(self): node = self._create_node() self.assertItemsEqual( [node], __unit__.breadth_first(node, self.CHILDREN_FUNC))
def test_start__some_object(self): obj = object() bfs = __unit__.breadth_first(obj, expand=functions.empty()) self.assertItemsEqual([obj], bfs)
def test_expand__some_object(self): with self.assertRaises(TypeError): __unit__.breadth_first(self._create_node(), expand=object())
def test_expand__none(self): with self.assertRaises(TypeError): __unit__.breadth_first(self._create_node(), expand=None)
def test_start__single_node(self): node = self._create_node() self.assertItemsEqual([node], __unit__.breadth_first(node, self.CHILDREN_FUNC))
def test_start__path(self): graph = self._create_path(10) bfs = __unit__.breadth_first(graph, self.CHILDREN_FUNC) for i, node in enumerate(bfs, 0): self.assertEquals(i, node.value)