from __future__ in tests too

celery/tests/test_buckets.py

@@ -1,3 +1,4 @@
+from __future__ import generators
 import os
 import sys
 sys.path.insert(0, os.getcwd())

celery/tests/test_log.py

@@ -1,4 +1,4 @@
-from __future__ import with_statement
+from __future__ import with_statement, generators
 
 import os
 import sys

celery/tests/test_result.py

@@ -1,3 +1,4 @@
+from __future__ import generators
 import unittest
 
 from celery import states

celery/tests/test_task_http.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from __future__ import with_statement
+from __future__ import with_statement, generators
 
 import sys
 import logging

celery/tests/test_worker_scheduler.py

@@ -1,3 +1,4 @@
+from __future__ import generators
 import unittest
 from Queue import Queue, Empty
 from datetime import datetime, timedelta

celery/tests/utils.py

@@ -1,4 +1,4 @@
-from __future__ import with_statement
+from __future__ import with_statement, generators
 
 import os
 import sys