def setUp(self): """Prepare model test fixture.""" try: new_attrs = {} new_attrs.update(self.attrs) new_attrs.update(self.do_get_dependencies()) self.obj = self.klass(**new_attrs) #pylint: disable=not-callable DBSession.add(self.obj) DBSession.flush() return self.obj except Exception: DBSession.rollback() raise
def tearDown(self): """Finish model test fixture.""" DBSession.rollback()