gh-131178: Add tests for pickle command-line interface (#131275)

Co-authored-by: Bénédikt Tran <10796600+picnixz@users.noreply.github.com>
This commit is contained in:
Semyon Moroz 2025-04-06 19:50:32 +04:00 committed by GitHub
parent dbfc6a417a
commit f247e1d04c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 64 additions and 5 deletions

View file

@ -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()