diff options
Diffstat (limited to 'test/testbase.py')
-rw-r--r-- | test/testbase.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/test/testbase.py b/test/testbase.py index 7d981509d..bc5153af0 100644 --- a/test/testbase.py +++ b/test/testbase.py @@ -205,6 +205,8 @@ class AssertMixin(PersistTest): self.assert_(db.sql_count == count, "desired statement count %d does not match %d" % (count, db.sql_count)) class ORMTest(AssertMixin): + keep_mappers = False + keep_data = False def setUpAll(self): global metadata metadata = BoundMetaData(db) @@ -217,10 +219,11 @@ class ORMTest(AssertMixin): def tearDownAll(self): metadata.drop_all() def tearDown(self): - clear_mappers() - for t in metadata.table_iterator(reverse=True): - t.delete().execute().close() - + if not self.keep_mappers: + clear_mappers() + if not self.keep_data: + for t in metadata.table_iterator(reverse=True): + t.delete().execute().close() class EngineAssert(proxy.BaseProxyEngine): """decorates a SQLEngine object to match the incoming queries against a set of assertions.""" |