Add an option to choose the IO module under test (allows to bench

e.g. the pure Python implementation in _pyio).
This commit is contained in:
Antoine Pitrou 2010-09-06 12:36:55 +00:00
parent 13b55291ac
commit 92f87f7c49
1 changed files with 6 additions and 0 deletions

View File

@ -427,6 +427,9 @@ def main():
action="store", dest="newlines", default='lf', action="store", dest="newlines", default='lf',
help="line endings for text tests " help="line endings for text tests "
"(one of: {lf (default), cr, crlf, all})") "(one of: {lf (default), cr, crlf, all})")
parser.add_option("-m", "--io-module",
action="store", dest="io_module", default=None,
help="io module to test (default: builtin open())")
options, args = parser.parse_args() options, args = parser.parse_args()
if args: if args:
parser.error("unexpected arguments") parser.error("unexpected arguments")
@ -451,6 +454,9 @@ def main():
if options.encoding: if options.encoding:
TEXT_ENCODING = options.encoding TEXT_ENCODING = options.encoding
if options.io_module:
globals()['open'] = __import__(options.io_module, {}, {}, ['open']).open
prepare_files() prepare_files()
run_all_tests(test_options) run_all_tests(test_options)