Using inheritance
import cPickle import pickle import unittest class TestPickleBase: def test_dumps_loads(self): self.assertEqual(self.pickle_impl.loads(self.pickle_impl.dumps('a')), 'a') class TestCPickle(unittest.TestCase, TestPickleBase): pickle_impl = cPickle class TestPickle(unittest.TestCase, TestPickleBase): pickle_impl = pickle if __name__ == '__main__': unittest.main()
import cPickle import pickle import pytest @pytest.mark.parametrize('pickle_impl', [pickle, cPickle]) def test_dumps_loads(pickle_impl): assert pickle_impl.loads(pickle_impl.dumps('a')) == 'a'
import cPickle import pickle import pytest @pytest.fixture(params=[pickle, cPickle]) def pickle_impl(request): return request.param def test_dumps_loads(pickle_impl): assert pickle_impl.loads(pickle_impl.dumps('a')) == 'a'
source share