blob: a4879179249176a55b9880168401215c65e579ce [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`timeit` --- Measure execution time of small code snippets
2===============================================================
3
4.. module:: timeit
5 :synopsis: Measure the execution time of small code snippets.
6
7
Georg Brandl116aa622007-08-15 14:28:22 +00008.. index::
9 single: Benchmarking
10 single: Performance
11
Raymond Hettingera1993682011-01-27 01:20:32 +000012**Source code:** :source:`Lib/timeit.py`
13
14--------------
15
Georg Brandl116aa622007-08-15 14:28:22 +000016This module provides a simple way to time small bits of Python code. It has both
Ezio Melottid0fe3e52012-10-02 05:35:39 +030017a :ref:`command-line-interface` as well as a :ref:`callable <python-interface>`
18one. It avoids a number of common traps for measuring execution times.
19See also Tim Peters' introduction to the "Algorithms" chapter in the *Python
20Cookbook*, published by O'Reilly.
Georg Brandl116aa622007-08-15 14:28:22 +000021
22
Ezio Melottid0fe3e52012-10-02 05:35:39 +030023Basic Examples
24--------------
Georg Brandl116aa622007-08-15 14:28:22 +000025
Ezio Melottid0fe3e52012-10-02 05:35:39 +030026The following example shows how the :ref:`command-line-interface`
27can be used to compare three different expressions:
Georg Brandl116aa622007-08-15 14:28:22 +000028
Ezio Melottid0fe3e52012-10-02 05:35:39 +030029.. code-block:: sh
Georg Brandl116aa622007-08-15 14:28:22 +000030
Ezio Melottid0fe3e52012-10-02 05:35:39 +030031 $ python -m timeit '"-".join(str(n) for n in range(100))'
32 10000 loops, best of 3: 40.3 usec per loop
33 $ python -m timeit '"-".join([str(n) for n in range(100)])'
34 10000 loops, best of 3: 33.4 usec per loop
35 $ python -m timeit '"-".join(map(str, range(100)))'
36 10000 loops, best of 3: 25.2 usec per loop
Georg Brandl116aa622007-08-15 14:28:22 +000037
Ezio Melottid0fe3e52012-10-02 05:35:39 +030038This can be achieved from the :ref:`python-interface` with::
39
40 >>> import timeit
41 >>> timeit.timeit('"-".join(str(n) for n in range(100))', number=10000)
42 0.8187260627746582
43 >>> timeit.timeit('"-".join([str(n) for n in range(100)])', number=10000)
44 0.7288308143615723
45 >>> timeit.timeit('"-".join(map(str, range(100)))', number=10000)
46 0.5858950614929199
47
48Note however that :mod:`timeit` will automatically determine the number of
49repetitions only when the command-line interface is used. In the
50:ref:`timeit-examples` section you can find more advanced examples.
Georg Brandl116aa622007-08-15 14:28:22 +000051
52
Ezio Melottid0fe3e52012-10-02 05:35:39 +030053.. _python-interface:
Georg Brandl116aa622007-08-15 14:28:22 +000054
Ezio Melottid0fe3e52012-10-02 05:35:39 +030055Python Interface
56----------------
Georg Brandl116aa622007-08-15 14:28:22 +000057
Ezio Melottid0fe3e52012-10-02 05:35:39 +030058The module defines three convenience functions and a public class:
Georg Brandl116aa622007-08-15 14:28:22 +000059
60
Ezio Melottid0fe3e52012-10-02 05:35:39 +030061.. function:: timeit(stmt='pass', setup='pass', timer=<default timer>, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +000062
Ezio Melottid0fe3e52012-10-02 05:35:39 +030063 Create a :class:`Timer` instance with the given statement, *setup* code and
64 *timer* function and run its :meth:`.timeit` method with *number* executions.
Georg Brandl116aa622007-08-15 14:28:22 +000065
66
Ezio Melottid0fe3e52012-10-02 05:35:39 +030067.. function:: repeat(stmt='pass', setup='pass', timer=<default timer>, repeat=3, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +000068
Ezio Melottid0fe3e52012-10-02 05:35:39 +030069 Create a :class:`Timer` instance with the given statement, *setup* code and
70 *timer* function and run its :meth:`.repeat` method with the given *repeat*
71 count and *number* executions.
Sandro Tosie6c34622012-04-24 18:11:46 +020072
73
74.. function:: default_timer()
75
Georg Brandl67c14442012-05-01 11:59:36 +020076 The default timer, which is always :func:`time.perf_counter`.
Sandro Tosie6c34622012-04-24 18:11:46 +020077
Ezio Melotti44437622012-10-02 06:01:16 +030078 .. versionchanged:: 3.3
79 :func:`time.perf_counter` is now the default timer.
Georg Brandl116aa622007-08-15 14:28:22 +000080
Georg Brandl116aa622007-08-15 14:28:22 +000081
Georg Brandl116aa622007-08-15 14:28:22 +000082.. class:: Timer(stmt='pass', setup='pass', timer=<timer function>)
Georg Brandl116aa622007-08-15 14:28:22 +000083
Georg Brandl116aa622007-08-15 14:28:22 +000084 Class for timing execution speed of small code snippets.
85
Ezio Melottid0fe3e52012-10-02 05:35:39 +030086 The constructor takes a statement to be timed, an additional statement used
87 for setup, and a timer function. Both statements default to ``'pass'``;
88 the timer function is platform-dependent (see the module doc string).
89 *stmt* and *setup* may also contain multiple statements separated by ``;``
90 or newlines, as long as they don't contain multi-line string literals.
Georg Brandl116aa622007-08-15 14:28:22 +000091
Ezio Melottid0fe3e52012-10-02 05:35:39 +030092 To measure the execution time of the first statement, use the :meth:`.timeit`
93 method. The :meth:`.repeat` method is a convenience to call :meth:`.timeit`
Georg Brandl116aa622007-08-15 14:28:22 +000094 multiple times and return a list of results.
95
96 The *stmt* and *setup* parameters can also take objects that are callable
Ezio Melottid0fe3e52012-10-02 05:35:39 +030097 without arguments. This will embed calls to them in a timer function that
Ezio Melottia3ccb232012-09-20 06:13:38 +030098 will then be executed by :meth:`.timeit`. Note that the timing overhead is a
Georg Brandl116aa622007-08-15 14:28:22 +000099 little larger in this case because of the extra function calls.
Georg Brandl116aa622007-08-15 14:28:22 +0000100
Georg Brandl116aa622007-08-15 14:28:22 +0000101
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300102 .. method:: Timer.timeit(number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +0000103
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300104 Time *number* executions of the main statement. This executes the setup
105 statement once, and then returns the time it takes to execute the main
106 statement a number of times, measured in seconds as a float.
107 The argument is the number of times through the loop, defaulting to one
108 million. The main statement, the setup statement and the timer function
109 to be used are passed to the constructor.
Georg Brandl116aa622007-08-15 14:28:22 +0000110
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300111 .. note::
Georg Brandl116aa622007-08-15 14:28:22 +0000112
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300113 By default, :meth:`.timeit` temporarily turns off :term:`garbage
114 collection` during the timing. The advantage of this approach is that
115 it makes independent timings more comparable. This disadvantage is
116 that GC may be an important component of the performance of the
117 function being measured. If so, GC can be re-enabled as the first
118 statement in the *setup* string. For example::
Georg Brandl116aa622007-08-15 14:28:22 +0000119
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300120 timeit.Timer('for i in range(10): oct(i)', 'gc.enable()').timeit()
Georg Brandl116aa622007-08-15 14:28:22 +0000121
122
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300123 .. method:: Timer.repeat(repeat=3, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +0000124
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300125 Call :meth:`.timeit` a few times.
Georg Brandl116aa622007-08-15 14:28:22 +0000126
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300127 This is a convenience function that calls the :meth:`.timeit` repeatedly,
128 returning a list of results. The first argument specifies how many times
129 to call :meth:`.timeit`. The second argument specifies the *number*
130 argument for :meth:`.timeit`.
Georg Brandl116aa622007-08-15 14:28:22 +0000131
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300132 .. note::
Georg Brandl116aa622007-08-15 14:28:22 +0000133
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300134 It's tempting to calculate mean and standard deviation from the result
135 vector and report these. However, this is not very useful.
136 In a typical case, the lowest value gives a lower bound for how fast
137 your machine can run the given code snippet; higher values in the
138 result vector are typically not caused by variability in Python's
139 speed, but by other processes interfering with your timing accuracy.
140 So the :func:`min` of the result is probably the only number you
141 should be interested in. After that, you should look at the entire
142 vector and apply common sense rather than statistics.
Georg Brandl116aa622007-08-15 14:28:22 +0000143
144
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300145 .. method:: Timer.print_exc(file=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000146
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300147 Helper to print a traceback from the timed code.
Georg Brandl116aa622007-08-15 14:28:22 +0000148
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300149 Typical use::
Georg Brandl116aa622007-08-15 14:28:22 +0000150
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300151 t = Timer(...) # outside the try/except
152 try:
153 t.timeit(...) # or t.repeat(...)
154 except:
155 t.print_exc()
Georg Brandl116aa622007-08-15 14:28:22 +0000156
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300157 The advantage over the standard traceback is that source lines in the
158 compiled template will be displayed. The optional *file* argument directs
159 where the traceback is sent; it defaults to :data:`sys.stderr`.
Georg Brandl116aa622007-08-15 14:28:22 +0000160
161
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300162.. _command-line-interface:
Georg Brandl116aa622007-08-15 14:28:22 +0000163
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300164Command-Line Interface
Georg Brandl116aa622007-08-15 14:28:22 +0000165----------------------
166
167When called as a program from the command line, the following form is used::
168
169 python -m timeit [-n N] [-r N] [-s S] [-t] [-c] [-h] [statement ...]
170
Éric Araujo713d3032010-11-18 16:38:46 +0000171Where the following options are understood:
Georg Brandl116aa622007-08-15 14:28:22 +0000172
Éric Araujo713d3032010-11-18 16:38:46 +0000173.. program:: timeit
174
175.. cmdoption:: -n N, --number=N
176
Georg Brandl116aa622007-08-15 14:28:22 +0000177 how many times to execute 'statement'
178
Éric Araujo713d3032010-11-18 16:38:46 +0000179.. cmdoption:: -r N, --repeat=N
180
Georg Brandl116aa622007-08-15 14:28:22 +0000181 how many times to repeat the timer (default 3)
182
Éric Araujo713d3032010-11-18 16:38:46 +0000183.. cmdoption:: -s S, --setup=S
Georg Brandl116aa622007-08-15 14:28:22 +0000184
Éric Araujo713d3032010-11-18 16:38:46 +0000185 statement to be executed once initially (default ``pass``)
186
Georg Brandl67c14442012-05-01 11:59:36 +0200187.. cmdoption:: -p, --process
188
189 measure process time, not wallclock time, using :func:`time.process_time`
190 instead of :func:`time.perf_counter`, which is the default
191
192 .. versionadded:: 3.3
193
Éric Araujo713d3032010-11-18 16:38:46 +0000194.. cmdoption:: -t, --time
195
Georg Brandl67c14442012-05-01 11:59:36 +0200196 use :func:`time.time` (deprecated)
Georg Brandl116aa622007-08-15 14:28:22 +0000197
Éric Araujo713d3032010-11-18 16:38:46 +0000198.. cmdoption:: -c, --clock
199
Georg Brandl67c14442012-05-01 11:59:36 +0200200 use :func:`time.clock` (deprecated)
Georg Brandl116aa622007-08-15 14:28:22 +0000201
Éric Araujo713d3032010-11-18 16:38:46 +0000202.. cmdoption:: -v, --verbose
203
Georg Brandl116aa622007-08-15 14:28:22 +0000204 print raw timing results; repeat for more digits precision
205
Éric Araujo713d3032010-11-18 16:38:46 +0000206.. cmdoption:: -h, --help
207
Georg Brandl116aa622007-08-15 14:28:22 +0000208 print a short usage message and exit
209
210A multi-line statement may be given by specifying each line as a separate
211statement argument; indented lines are possible by enclosing an argument in
212quotes and using leading spaces. Multiple :option:`-s` options are treated
213similarly.
214
215If :option:`-n` is not given, a suitable number of loops is calculated by trying
216successive powers of 10 until the total time is at least 0.2 seconds.
217
Georg Brandl67c14442012-05-01 11:59:36 +0200218:func:`default_timer` measurements can be affected by other programs running on
219the same machine, so the best thing to do when accurate timing is necessary is
220to repeat the timing a few times and use the best time. The :option:`-r`
221option is good for this; the default of 3 repetitions is probably enough in
222most cases. You can use :func:`time.process_time` to measure CPU time.
Georg Brandl116aa622007-08-15 14:28:22 +0000223
224.. note::
225
226 There is a certain baseline overhead associated with executing a pass statement.
227 The code here doesn't try to hide it, but you should be aware of it. The
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300228 baseline overhead can be measured by invoking the program without arguments,
229 and it might differ between Python versions.
Georg Brandl116aa622007-08-15 14:28:22 +0000230
Georg Brandl116aa622007-08-15 14:28:22 +0000231
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300232.. _timeit-examples:
Georg Brandl116aa622007-08-15 14:28:22 +0000233
234Examples
235--------
236
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300237It is possible to provide a setup statement that is executed only once at the beginning:
238
239.. code-block:: sh
240
241 $ python -m timeit -s 'text = "sample string"; char = "g"' 'char in text'
242 10000000 loops, best of 3: 0.0877 usec per loop
243 $ python -m timeit -s 'text = "sample string"; char = "g"' 'text.find(char)'
244 1000000 loops, best of 3: 0.342 usec per loop
245
246::
247
248 >>> import timeit
249 >>> timeit.timeit('char in text', setup='text = "sample string"; char = "g"')
250 0.41440500499993504
251 >>> timeit.timeit('text.find(char)', setup='text = "sample string"; char = "g"')
252 1.7246671520006203
253
254The same can be done using the :class:`Timer` class and its methods::
255
256 >>> import timeit
257 >>> t = timeit.Timer('char in text', setup='text = "sample string"; char = "g"')
258 >>> t.timeit()
259 0.3955516149999312
260 >>> t.repeat()
261 [0.40193588800002544, 0.3960157959998014, 0.39594301399984033]
262
263
264The following examples show how to time expressions that contain multiple lines.
265Here we compare the cost of using :func:`hasattr` vs. :keyword:`try`/:keyword:`except`
266to test for missing and present object attributes:
267
268.. code-block:: sh
Georg Brandl116aa622007-08-15 14:28:22 +0000269
Senthil Kumaran2e015352011-08-06 13:37:04 +0800270 $ python -m timeit 'try:' ' str.__bool__' 'except AttributeError:' ' pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000271 100000 loops, best of 3: 15.7 usec per loop
Senthil Kumaran2e015352011-08-06 13:37:04 +0800272 $ python -m timeit 'if hasattr(str, "__bool__"): pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000273 100000 loops, best of 3: 4.26 usec per loop
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300274
Senthil Kumaran2e015352011-08-06 13:37:04 +0800275 $ python -m timeit 'try:' ' int.__bool__' 'except AttributeError:' ' pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000276 1000000 loops, best of 3: 1.43 usec per loop
Senthil Kumaran2e015352011-08-06 13:37:04 +0800277 $ python -m timeit 'if hasattr(int, "__bool__"): pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000278 100000 loops, best of 3: 2.23 usec per loop
279
280::
281
282 >>> import timeit
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300283 >>> # attribute is missing
Georg Brandl116aa622007-08-15 14:28:22 +0000284 >>> s = """\
285 ... try:
286 ... str.__bool__
287 ... except AttributeError:
288 ... pass
289 ... """
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300290 >>> timeit.timeit(stmt=s, number=100000)
291 0.9138244460009446
292 >>> s = "if hasattr(str, '__bool__'): pass"
293 >>> timeit.timeit(stmt=s, number=100000)
294 0.5829014980008651
295 >>>
296 >>> # attribute is present
Georg Brandl116aa622007-08-15 14:28:22 +0000297 >>> s = """\
298 ... try:
299 ... int.__bool__
300 ... except AttributeError:
301 ... pass
302 ... """
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300303 >>> timeit.timeit(stmt=s, number=100000)
304 0.04215312199994514
305 >>> s = "if hasattr(int, '__bool__'): pass"
306 >>> timeit.timeit(stmt=s, number=100000)
307 0.08588060699912603
308
Georg Brandl116aa622007-08-15 14:28:22 +0000309
310To give the :mod:`timeit` module access to functions you define, you can pass a
Ezio Melottia3ccb232012-09-20 06:13:38 +0300311*setup* parameter which contains an import statement::
Georg Brandl116aa622007-08-15 14:28:22 +0000312
313 def test():
Senthil Kumaran2e015352011-08-06 13:37:04 +0800314 """Stupid test function"""
Collin Winterc79461b2007-09-01 23:34:30 +0000315 L = [i for i in range(100)]
Georg Brandl116aa622007-08-15 14:28:22 +0000316
Senthil Kumaran2e015352011-08-06 13:37:04 +0800317 if __name__ == '__main__':
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300318 import timeit
319 print(timeit.timeit("test()", setup="from __main__ import test"))