def tearDown(self): """Undo all the state we have.""" rmtree(self.testdata_dir) rmtree(self.virtualenv_dir) DB.drop_tables(ORM_OBJECTS, cascade=True) DB.drop_tables([MetadataSystem]) DB.close()
def tearDownClass(cls): """Close the database connection.""" DB.close()
def setUp(self): """Setup the database.""" self._models = self.dependent_cls() DB.create_tables(self._models)
def tearDown(self): """Tear down the database.""" DB.drop_tables(self._models)
def setUpClass(cls): """Connect to the database.""" if not DB.is_closed(): # pragma: no cover this is just in case DB.close() DB.connect()