def setup(self): if self.run_define_tables == 'each': self.tables.clear() drop_all_tables(self.metadata) self.metadata.clear() self.define_tables(self.metadata) self.metadata.create_all() self.tables.update(self.metadata.tables) if self.run_setup_classes == 'each': self.classes.clear() baseline = subclasses(BasicEntity) self.setup_classes() self._register_new_class_artifacts(baseline) if self.run_setup_mappers == 'each': baseline = subclasses(BasicEntity) self.setup_mappers() self._register_new_class_artifacts(baseline) if self.run_inserts == 'each': self._load_fixtures() self.insert_data()
def teardown_class(cls): for cl in cls.classes.values(): cls.unregister_class(cl) ORMTest.teardown_class() drop_all_tables(cls.metadata) cls.metadata.bind = None