celery/celery/tests/test_utils/test_serialization.py

from __future__ import absolute_import
from __future__ import with_statement

import sys

from celery.tests.utils import Case, mask_modules


class test_AAPickle(Case):

    def test_no_cpickle(self):
        prev = sys.modules.pop("celery.utils.serialization", None)
        try:
            with mask_modules("cPickle"):
                from celery.utils.serialization import pickle
                import pickle as orig_pickle
                self.assertIs(pickle.dumps, orig_pickle.dumps)
        finally:
            sys.modules["celery.utils.serialization"] = prev