From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 8 Nov 2014 16:24:47 +0000 (-0500)
Subject: move this
X-Git-Tag: rel_0_7_0~49
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=e2c46aa37a48bda3ffc1defee79b69efc8445ad3;p=thirdparty%2Fsqlalchemy%2Falembic.git

move this
---

diff --git a/tests/test_batch.py b/tests/test_batch.py
index c3df74d8..c4a309e2 100644
--- 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,