import unittest from app.App import App class test_OptimizationPostmanRepo(unittest.TestCase): @classmethod def setUpClass(cls): App.init() def test_all(self): optimizationPostmans = App.repos.optimizationPostmanRepo.getAllConfirmed() self.assertGreater(len(optimizationPostmans), 0) optimization = App.repos.optimizationPostmanRepo.get(posta=optimizationPostmans[0].id) self.assertIn(optimization, optimizationPostmans)