def test_is_last_page(self): menu = Menu(id=1, parent=None, weight=1) self.session.add(menu) page_1 = Page(id=2, parent=menu, weight=1) self.session.add(page_1) self.assertEqual(True, Page.is_last_page(self.session)) page_2 = Page(id=3, parent=menu, weight=2) self.session.add(page_2) self.assertEqual(False, Page.is_last_page(self.session))