Remove schema mgnt warning for all cqlengine tests

aholmberg · aholmberg · commit 7f03f079b9e0 · 2015-02-12T15:12:42.000-06:00
diff --git a/tests/integration/cqlengine/__init__.py b/tests/integration/cqlengine/__init__.py
@@ -12,16 +12,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import warnings
 
 from cassandra.cqlengine import connection
-from cassandra.cqlengine.management import create_keyspace_simple
+from cassandra.cqlengine.management import create_keyspace_simple, CQLENG_ALLOW_SCHEMA_MANAGEMENT
 
 from tests.integration import use_single_node, PROTOCOL_VERSION
 
 
 def setup_package():
     warnings.simplefilter('always')  # for testing warnings, make sure all are let through
+    os.environ[CQLENG_ALLOW_SCHEMA_MANAGEMENT] = '1'
 
     use_single_node()
 
diff --git a/tests/integration/cqlengine/management/__init__.py b/tests/integration/cqlengine/management/__init__.py
@@ -12,9 +12,3 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
-
-from cassandra.cqlengine.management import CQLENG_ALLOW_SCHEMA_MANAGEMENT
-
-def setup_package():
-    os.environ[CQLENG_ALLOW_SCHEMA_MANAGEMENT] = '1'
