diff options
| author | Semyon Moroz <donbarbos@proton.me> | 2025-04-06 15:50:32 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-04-06 15:50:32 (GMT) |
| commit | f247e1d04c91b5a353c499487248dda93c151d2d (patch) | |
| tree | 3748614f0c6acf6f02981baf39754a37239fca49 /Lib/pickle.py | |
| parent | dbfc6a417a5db237ff16a8a75f09c1bd5d5bd349 (diff) | |
| download | cpython-f247e1d04c91b5a353c499487248dda93c151d2d.zip cpython-f247e1d04c91b5a353c499487248dda93c151d2d.tar.gz cpython-f247e1d04c91b5a353c499487248dda93c151d2d.tar.bz2 | |
gh-131178: Add tests for `pickle` command-line interface (#131275)
Co-authored-by: Bénédikt Tran <10796600+picnixz@users.noreply.github.com>
Diffstat (limited to 'Lib/pickle.py')
| -rw-r--r-- | Lib/pickle.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Lib/pickle.py b/Lib/pickle.py index efcdcbe..4fa3632 100644 --- a/Lib/pickle.py +++ b/Lib/pickle.py @@ -1907,7 +1907,7 @@ except ImportError: dump, dumps, load, loads = _dump, _dumps, _load, _loads -if __name__ == "__main__": +def _main(args=None): import argparse import pprint parser = argparse.ArgumentParser( @@ -1915,7 +1915,7 @@ if __name__ == "__main__": parser.add_argument( 'pickle_file', nargs='+', help='the pickle file') - args = parser.parse_args() + args = parser.parse_args(args) for fn in args.pickle_file: if fn == '-': obj = load(sys.stdin.buffer) @@ -1923,3 +1923,7 @@ if __name__ == "__main__": with open(fn, 'rb') as f: obj = load(f) pprint.pprint(obj) + + +if __name__ == "__main__": + _main() |