Commit 21290eac authored by Raymond Hettinger's avatar Raymond Hettinger

Add tests for pickletools.optimize().

parent 6bf51da9
import pickle
import pickletools
from test import test_support
test_support.run_doctest(pickletools)
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()
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment