move this

diff --git a/tests/test_batch.py b/tests/test_batch.py
index c3df74d8f78c70f482f6a71d9de79d7f76ef7276..c4a309e21428d76f73f2b4b6a113149b31ce4ecd 100644 (file)
--- a/tests/test_batch.py
+++ b/tests/test_batch.py
@@ -155,12 +155,12 @@ class BatchAPITest(TestBase):
 
         with mock.patch("alembic.operations.sa_schema") as mock_schema:
             yield batch
+        batch.impl.flush()
         self.mock_schema = mock_schema
 
     def test_drop_col(self):
         with self._fixture() as batch:
             batch.drop_column('q')
-            batch.impl.flush()
 
         eq_(
             batch.impl.operations.impl.mock_calls,
@@ -173,7 +173,6 @@ class BatchAPITest(TestBase):
 
         with self._fixture() as batch:
             batch.add_column(column)
-            batch.impl.flush()
 
         eq_(
             batch.impl.operations.impl.mock_calls,