From 92f87f7c49179e271e893898ec898f9908508313 Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Mon, 6 Sep 2010 12:36:55 +0000 Subject: [PATCH] Add an option to choose the IO module under test (allows to bench e.g. the pure Python implementation in _pyio). --- Tools/iobench/iobench.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/Tools/iobench/iobench.py b/Tools/iobench/iobench.py index 92b4ffa180c..b3bdd6a6709 100644 --- a/Tools/iobench/iobench.py +++ b/Tools/iobench/iobench.py @@ -427,6 +427,9 @@ def main(): action="store", dest="newlines", default='lf', help="line endings for text tests " "(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() if args: parser.error("unexpected arguments") @@ -451,6 +454,9 @@ def main(): if options.encoding: TEXT_ENCODING = options.encoding + if options.io_module: + globals()['open'] = __import__(options.io_module, {}, {}, ['open']).open + prepare_files() run_all_tests(test_options)