diff options
author | Fred Drake <fdrake@acm.org> | 2001-12-19 16:42:15 +0000 |
---|---|---|
committer | Fred Drake <fdrake@acm.org> | 2001-12-19 16:42:15 +0000 |
commit | e48480465607cdfa9680685bb3c0fb5b78d1f28d (patch) | |
tree | 378160b92552d772626c4e5e95c3aa539b3f0e4e /Lib/test/test_pickle.py | |
parent | 235a288c6c8e39bef46bfd3e8f56fa50ea17f620 (diff) | |
download | cpython-e48480465607cdfa9680685bb3c0fb5b78d1f28d.tar.gz |
Fix the test control support for the pickle & cPickle tests so the tests run
under regrtest.
Diffstat (limited to 'Lib/test/test_pickle.py')
-rw-r--r-- | Lib/test/test_pickle.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/Lib/test/test_pickle.py b/Lib/test/test_pickle.py index 06a554e05e..6265676533 100644 --- a/Lib/test/test_pickle.py +++ b/Lib/test/test_pickle.py @@ -1,7 +1,8 @@ import pickle +import test_support +import unittest from cStringIO import StringIO from pickletester import AbstractPickleTests, AbstractPickleModuleTests -from test_support import run_unittest class PickleTests(AbstractPickleTests, AbstractPickleModuleTests): @@ -28,6 +29,12 @@ class PicklerTests(AbstractPickleTests): u = pickle.Unpickler(f) return u.load() +def test_main(): + loader = unittest.TestLoader() + suite = unittest.TestSuite() + suite.addTest(loader.loadTestsFromTestCase(PickleTests)) + suite.addTest(loader.loadTestsFromTestCase(PicklerTests)) + test_support.run_suite(suite) + if __name__ == "__main__": - run_unittest(PickleTests) - run_unittest(PicklerTests) + test_main() |