From e2c46aa37a48bda3ffc1defee79b69efc8445ad3 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 8 Nov 2014 11:24:47 -0500 Subject: [PATCH] move this --- tests/test_batch.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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, -- 2.47.3