Fixing bug with test_queryset test collision

GregBestland · GregBestland · commit c52cdba0e6f9 · 2016-04-05T09:59:59.000-05:00
diff --git a/tests/integration/cqlengine/query/test_queryset.py b/tests/integration/cqlengine/query/test_queryset.py
@@ -978,15 +978,13 @@ class DBFieldModelMixed2(Model):
 
 class TestModelQueryWithDBField(BaseCassEngTestCase):
 
-    @classmethod
-    def setUpClass(cls):
+    def setUp(cls):
         super(TestModelQueryWithDBField, cls).setUpClass()
         cls.model_list = [DBFieldModel, DBFieldModelMixed1, DBFieldModelMixed2]
         for model in cls.model_list:
             sync_table(model)
 
-    @classmethod
-    def tearDownClass(cls):
+    def tearDown(cls):
         super(TestModelQueryWithDBField, cls).tearDownClass()
         for model in cls.model_list:
             drop_table(model)
@@ -1120,8 +1118,6 @@ def test_db_field_names_used(self):
             self.assertTrue(value not in str(b2.queries[0]))
 
 
-
-
 class TestModelSmall(Model):
 
     test_id = columns.Integer(primary_key=True)
