diff --git a/test/test_persistence.py b/test/test_persistence.py index f0792bb..9b6a96b 100644 --- a/test/test_persistence.py +++ b/test/test_persistence.py @@ -19,6 +19,10 @@ class DatabaseTestCase(unittest.TestCase): for row in TEST_DATA: self.tbl.insert(row) + def tearDown(self): + for table in self.db.tables: + self.db[table].drop() + def test_valid_database_url(self): assert self.db.url, os.environ['DATABASE_URL']