Staging
v0.8.1
Revision 4ed35fc4f31ebbdb25ba5e1291de77f4095c33d0 authored by Martin Panter on 10 October 2015, 10:52:35 UTC, committed by Martin Panter on 10 October 2015, 10:52:35 UTC
1 parent dab305e
Raw File
test_pickletools.py
import pickle
import pickletools
from test import test_support
from test.pickletester import AbstractPickleTests
from test.pickletester import AbstractPickleModuleTests

class OptimizedPickleTests(AbstractPickleTests, AbstractPickleModuleTests):

    def dumps(self, arg, proto=0, fast=0):
        return pickletools.optimize(pickle.dumps(arg, proto))

    def loads(self, buf):
        return pickle.loads(buf)

    module = pickle
    error = KeyError

def test_main():
    test_support.run_unittest(OptimizedPickleTests)
    test_support.run_doctest(pickletools)


if __name__ == "__main__":
    test_main()
back to top