def test_toplevel(self): assert set(get_all_subclasses(self.Base)) == set([ self.BranchI, self.BranchII, self.BranchIA, self.BranchIB, self.BranchIIA, self.BranchIIB ])
def test_bottom_level(self): assert get_all_subclasses(self.BranchIIB) == []
def test_one_inheritance(self): assert set(get_all_subclasses(self.BranchII)) == set( [self.BranchIIA, self.BranchIIB])
def get_all_pkg_managers(): return dict([(obj.__name__.lower(), obj) for obj in get_all_subclasses(PkgMgr) if obj not in (CLIMgr, LibMgr)])