Blob Blame History Raw
Index: sqlalchemy-migrate-0.7.1/migrate/tests/versioning/test_shell.py
===================================================================
--- sqlalchemy-migrate-0.7.1.orig/migrate/tests/versioning/test_shell.py
+++ sqlalchemy-migrate-0.7.1/migrate/tests/versioning/test_shell.py
@@ -482,8 +482,18 @@ class TestShellDatabase(Shell, DB):
         self.assertTrue("tables missing from database: tmp_account_rundiffs" in result.stdout)
 
         # Test Deprecation
+        if 'PYTHONWARNINGS' in self.env.environ:
+            warnings = self.env.environ['PYTHONWARNINGS']
+        else:
+            warnings = None
+        self.env.environ['PYTHONWARNINGS'] = 'default'
         result = self.env.run('migrate compare_model_to_db %s %s --model=%s' \
             % (self.url, repos_path, model_module.replace(":", ".")), expect_error=True)
+        if warnings == None:
+            del(self.env.environ['PYTHONWARNINGS'])
+        else:
+            self.env.environ['PYTHONWARNINGS'] = warnings
+
         self.assertEqual(result.returncode, 0)
         self.assertTrue("tables missing from database: tmp_account_rundiffs" in result.stdout)