def test_multiple_levels_deep(self, category): category_2 = Category(category.site, 'pages/category/category_2') category_3 = Category(category.site, 'pages/category/category_3') category.categories = [category_2] category_2.categories = [category_3] category_3.categories = [] assert category.sub_categories == [category_2, category_3]
def test_one_level_deep(self, category): category_2 = Category(category.site, 'pages/category/category_2') category.categories = [category_2] category_2.categories = [] assert category.sub_categories == [category_2]