def test_connection_state(self): conn = test_db.get_conn() self.assertFalse(test_db.is_closed()) test_db.close() self.assertTrue(test_db.is_closed()) conn = test_db.get_conn() self.assertFalse(test_db.is_closed())
def tearDown(self): super(TestFlaskDB, self).tearDown() if not test_db.is_closed(): test_db.close() test_db.connect()