blob: 19b5e4e7db97a83312da4c88172503c97b3b91e9 [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 Melotti591176e2014-08-04 17:01:16 +030031 $ python3 -m timeit '"-".join(str(n) for n in range(100))'
32 10000 loops, best of 3: 30.2 usec per loop
33 $ python3 -m timeit '"-".join([str(n) for n in range(100)])'
34 10000 loops, best of 3: 27.5 usec per loop
35 $ python3 -m timeit '"-".join(map(str, range(100)))'
36 10000 loops, best of 3: 23.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)
Ezio Melotti591176e2014-08-04 17:01:16 +030042 0.3018611848820001
Ezio Melottid0fe3e52012-10-02 05:35:39 +030043 >>> timeit.timeit('"-".join([str(n) for n in range(100)])', number=10000)
Ezio Melotti591176e2014-08-04 17:01:16 +030044 0.2727368790656328
Ezio Melottid0fe3e52012-10-02 05:35:39 +030045 >>> timeit.timeit('"-".join(map(str, range(100)))', number=10000)
Ezio Melotti591176e2014-08-04 17:01:16 +030046 0.23702679807320237
47
Ezio Melottid0fe3e52012-10-02 05:35:39 +030048
49Note however that :mod:`timeit` will automatically determine the number of
50repetitions only when the command-line interface is used. In the
51:ref:`timeit-examples` section you can find more advanced examples.
Georg Brandl116aa622007-08-15 14:28:22 +000052
53
Ezio Melottid0fe3e52012-10-02 05:35:39 +030054.. _python-interface:
Georg Brandl116aa622007-08-15 14:28:22 +000055
Ezio Melottid0fe3e52012-10-02 05:35:39 +030056Python Interface
57----------------
Georg Brandl116aa622007-08-15 14:28:22 +000058
Ezio Melottid0fe3e52012-10-02 05:35:39 +030059The module defines three convenience functions and a public class:
Georg Brandl116aa622007-08-15 14:28:22 +000060
61
Ezio Melottid0fe3e52012-10-02 05:35:39 +030062.. function:: timeit(stmt='pass', setup='pass', timer=<default timer>, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +000063
Ezio Melottid0fe3e52012-10-02 05:35:39 +030064 Create a :class:`Timer` instance with the given statement, *setup* code and
65 *timer* function and run its :meth:`.timeit` method with *number* executions.
Georg Brandl116aa622007-08-15 14:28:22 +000066
Andrew Kuchling44da19a2014-04-14 13:39:43 -040067 .. note::
68
69 Because :meth:`.timeit` is executing *stmt*, placing a return statement
70 in *stmt* will prevent :meth:`.timeit` from returning execution time.
71 It will instead return the data specified by your return statement.
72
Georg Brandl116aa622007-08-15 14:28:22 +000073
Ezio Melottid0fe3e52012-10-02 05:35:39 +030074.. function:: repeat(stmt='pass', setup='pass', timer=<default timer>, repeat=3, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +000075
Ezio Melottid0fe3e52012-10-02 05:35:39 +030076 Create a :class:`Timer` instance with the given statement, *setup* code and
77 *timer* function and run its :meth:`.repeat` method with the given *repeat*
78 count and *number* executions.
Sandro Tosie6c34622012-04-24 18:11:46 +020079
80
81.. function:: default_timer()
82
Georg Brandl67c14442012-05-01 11:59:36 +020083 The default timer, which is always :func:`time.perf_counter`.
Sandro Tosie6c34622012-04-24 18:11:46 +020084
Ezio Melotti44437622012-10-02 06:01:16 +030085 .. versionchanged:: 3.3
86 :func:`time.perf_counter` is now the default timer.
Georg Brandl116aa622007-08-15 14:28:22 +000087
Georg Brandl116aa622007-08-15 14:28:22 +000088
Georg Brandl116aa622007-08-15 14:28:22 +000089.. class:: Timer(stmt='pass', setup='pass', timer=<timer function>)
Georg Brandl116aa622007-08-15 14:28:22 +000090
Georg Brandl116aa622007-08-15 14:28:22 +000091 Class for timing execution speed of small code snippets.
92
Ezio Melottid0fe3e52012-10-02 05:35:39 +030093 The constructor takes a statement to be timed, an additional statement used
94 for setup, and a timer function. Both statements default to ``'pass'``;
95 the timer function is platform-dependent (see the module doc string).
96 *stmt* and *setup* may also contain multiple statements separated by ``;``
97 or newlines, as long as they don't contain multi-line string literals.
Georg Brandl116aa622007-08-15 14:28:22 +000098
Ezio Melottid0fe3e52012-10-02 05:35:39 +030099 To measure the execution time of the first statement, use the :meth:`.timeit`
100 method. The :meth:`.repeat` method is a convenience to call :meth:`.timeit`
Georg Brandl116aa622007-08-15 14:28:22 +0000101 multiple times and return a list of results.
102
103 The *stmt* and *setup* parameters can also take objects that are callable
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300104 without arguments. This will embed calls to them in a timer function that
Ezio Melottia3ccb232012-09-20 06:13:38 +0300105 will then be executed by :meth:`.timeit`. Note that the timing overhead is a
Georg Brandl116aa622007-08-15 14:28:22 +0000106 little larger in this case because of the extra function calls.
Georg Brandl116aa622007-08-15 14:28:22 +0000107
Georg Brandl116aa622007-08-15 14:28:22 +0000108
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300109 .. method:: Timer.timeit(number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +0000110
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300111 Time *number* executions of the main statement. This executes the setup
112 statement once, and then returns the time it takes to execute the main
113 statement a number of times, measured in seconds as a float.
114 The argument is the number of times through the loop, defaulting to one
115 million. The main statement, the setup statement and the timer function
116 to be used are passed to the constructor.
Georg Brandl116aa622007-08-15 14:28:22 +0000117
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300118 .. note::
Georg Brandl116aa622007-08-15 14:28:22 +0000119
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300120 By default, :meth:`.timeit` temporarily turns off :term:`garbage
121 collection` during the timing. The advantage of this approach is that
122 it makes independent timings more comparable. This disadvantage is
123 that GC may be an important component of the performance of the
124 function being measured. If so, GC can be re-enabled as the first
125 statement in the *setup* string. For example::
Georg Brandl116aa622007-08-15 14:28:22 +0000126
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300127 timeit.Timer('for i in range(10): oct(i)', 'gc.enable()').timeit()
Georg Brandl116aa622007-08-15 14:28:22 +0000128
129
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300130 .. method:: Timer.repeat(repeat=3, number=1000000)
Georg Brandl116aa622007-08-15 14:28:22 +0000131
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300132 Call :meth:`.timeit` a few times.
Georg Brandl116aa622007-08-15 14:28:22 +0000133
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300134 This is a convenience function that calls the :meth:`.timeit` repeatedly,
135 returning a list of results. The first argument specifies how many times
136 to call :meth:`.timeit`. The second argument specifies the *number*
137 argument for :meth:`.timeit`.
Georg Brandl116aa622007-08-15 14:28:22 +0000138
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300139 .. note::
Georg Brandl116aa622007-08-15 14:28:22 +0000140
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300141 It's tempting to calculate mean and standard deviation from the result
142 vector and report these. However, this is not very useful.
143 In a typical case, the lowest value gives a lower bound for how fast
144 your machine can run the given code snippet; higher values in the
145 result vector are typically not caused by variability in Python's
146 speed, but by other processes interfering with your timing accuracy.
147 So the :func:`min` of the result is probably the only number you
148 should be interested in. After that, you should look at the entire
149 vector and apply common sense rather than statistics.
Georg Brandl116aa622007-08-15 14:28:22 +0000150
151
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300152 .. method:: Timer.print_exc(file=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000153
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300154 Helper to print a traceback from the timed code.
Georg Brandl116aa622007-08-15 14:28:22 +0000155
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300156 Typical use::
Georg Brandl116aa622007-08-15 14:28:22 +0000157
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300158 t = Timer(...) # outside the try/except
159 try:
160 t.timeit(...) # or t.repeat(...)
Andrew Svetlov47395612012-11-02 22:07:26 +0200161 except Exception:
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300162 t.print_exc()
Georg Brandl116aa622007-08-15 14:28:22 +0000163
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300164 The advantage over the standard traceback is that source lines in the
165 compiled template will be displayed. The optional *file* argument directs
166 where the traceback is sent; it defaults to :data:`sys.stderr`.
Georg Brandl116aa622007-08-15 14:28:22 +0000167
168
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300169.. _command-line-interface:
Georg Brandl116aa622007-08-15 14:28:22 +0000170
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300171Command-Line Interface
Georg Brandl116aa622007-08-15 14:28:22 +0000172----------------------
173
174When called as a program from the command line, the following form is used::
175
176 python -m timeit [-n N] [-r N] [-s S] [-t] [-c] [-h] [statement ...]
177
Éric Araujo713d3032010-11-18 16:38:46 +0000178Where the following options are understood:
Georg Brandl116aa622007-08-15 14:28:22 +0000179
Éric Araujo713d3032010-11-18 16:38:46 +0000180.. program:: timeit
181
182.. cmdoption:: -n N, --number=N
183
Georg Brandl116aa622007-08-15 14:28:22 +0000184 how many times to execute 'statement'
185
Éric Araujo713d3032010-11-18 16:38:46 +0000186.. cmdoption:: -r N, --repeat=N
187
Georg Brandl116aa622007-08-15 14:28:22 +0000188 how many times to repeat the timer (default 3)
189
Éric Araujo713d3032010-11-18 16:38:46 +0000190.. cmdoption:: -s S, --setup=S
Georg Brandl116aa622007-08-15 14:28:22 +0000191
Éric Araujo713d3032010-11-18 16:38:46 +0000192 statement to be executed once initially (default ``pass``)
193
Georg Brandl67c14442012-05-01 11:59:36 +0200194.. cmdoption:: -p, --process
195
196 measure process time, not wallclock time, using :func:`time.process_time`
197 instead of :func:`time.perf_counter`, which is the default
198
199 .. versionadded:: 3.3
200
Éric Araujo713d3032010-11-18 16:38:46 +0000201.. cmdoption:: -t, --time
202
Georg Brandl67c14442012-05-01 11:59:36 +0200203 use :func:`time.time` (deprecated)
Georg Brandl116aa622007-08-15 14:28:22 +0000204
Éric Araujo713d3032010-11-18 16:38:46 +0000205.. cmdoption:: -c, --clock
206
Georg Brandl67c14442012-05-01 11:59:36 +0200207 use :func:`time.clock` (deprecated)
Georg Brandl116aa622007-08-15 14:28:22 +0000208
Éric Araujo713d3032010-11-18 16:38:46 +0000209.. cmdoption:: -v, --verbose
210
Georg Brandl116aa622007-08-15 14:28:22 +0000211 print raw timing results; repeat for more digits precision
212
Éric Araujo713d3032010-11-18 16:38:46 +0000213.. cmdoption:: -h, --help
214
Georg Brandl116aa622007-08-15 14:28:22 +0000215 print a short usage message and exit
216
217A multi-line statement may be given by specifying each line as a separate
218statement argument; indented lines are possible by enclosing an argument in
219quotes and using leading spaces. Multiple :option:`-s` options are treated
220similarly.
221
222If :option:`-n` is not given, a suitable number of loops is calculated by trying
223successive powers of 10 until the total time is at least 0.2 seconds.
224
Georg Brandl67c14442012-05-01 11:59:36 +0200225:func:`default_timer` measurements can be affected by other programs running on
226the same machine, so the best thing to do when accurate timing is necessary is
227to repeat the timing a few times and use the best time. The :option:`-r`
228option is good for this; the default of 3 repetitions is probably enough in
229most cases. You can use :func:`time.process_time` to measure CPU time.
Georg Brandl116aa622007-08-15 14:28:22 +0000230
231.. note::
232
233 There is a certain baseline overhead associated with executing a pass statement.
234 The code here doesn't try to hide it, but you should be aware of it. The
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300235 baseline overhead can be measured by invoking the program without arguments,
236 and it might differ between Python versions.
Georg Brandl116aa622007-08-15 14:28:22 +0000237
Georg Brandl116aa622007-08-15 14:28:22 +0000238
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300239.. _timeit-examples:
Georg Brandl116aa622007-08-15 14:28:22 +0000240
241Examples
242--------
243
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300244It is possible to provide a setup statement that is executed only once at the beginning:
245
246.. code-block:: sh
247
248 $ python -m timeit -s 'text = "sample string"; char = "g"' 'char in text'
249 10000000 loops, best of 3: 0.0877 usec per loop
250 $ python -m timeit -s 'text = "sample string"; char = "g"' 'text.find(char)'
251 1000000 loops, best of 3: 0.342 usec per loop
252
253::
254
255 >>> import timeit
256 >>> timeit.timeit('char in text', setup='text = "sample string"; char = "g"')
257 0.41440500499993504
258 >>> timeit.timeit('text.find(char)', setup='text = "sample string"; char = "g"')
259 1.7246671520006203
260
261The same can be done using the :class:`Timer` class and its methods::
262
263 >>> import timeit
264 >>> t = timeit.Timer('char in text', setup='text = "sample string"; char = "g"')
265 >>> t.timeit()
266 0.3955516149999312
267 >>> t.repeat()
268 [0.40193588800002544, 0.3960157959998014, 0.39594301399984033]
269
270
271The following examples show how to time expressions that contain multiple lines.
272Here we compare the cost of using :func:`hasattr` vs. :keyword:`try`/:keyword:`except`
273to test for missing and present object attributes:
274
275.. code-block:: sh
Georg Brandl116aa622007-08-15 14:28:22 +0000276
Senthil Kumaran2e015352011-08-06 13:37:04 +0800277 $ python -m timeit 'try:' ' str.__bool__' 'except AttributeError:' ' pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000278 100000 loops, best of 3: 15.7 usec per loop
Senthil Kumaran2e015352011-08-06 13:37:04 +0800279 $ python -m timeit 'if hasattr(str, "__bool__"): pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000280 100000 loops, best of 3: 4.26 usec per loop
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300281
Senthil Kumaran2e015352011-08-06 13:37:04 +0800282 $ python -m timeit 'try:' ' int.__bool__' 'except AttributeError:' ' pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000283 1000000 loops, best of 3: 1.43 usec per loop
Senthil Kumaran2e015352011-08-06 13:37:04 +0800284 $ python -m timeit 'if hasattr(int, "__bool__"): pass'
Georg Brandl116aa622007-08-15 14:28:22 +0000285 100000 loops, best of 3: 2.23 usec per loop
286
287::
288
289 >>> import timeit
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300290 >>> # attribute is missing
Georg Brandl116aa622007-08-15 14:28:22 +0000291 >>> s = """\
292 ... try:
293 ... str.__bool__
294 ... except AttributeError:
295 ... pass
296 ... """
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300297 >>> timeit.timeit(stmt=s, number=100000)
298 0.9138244460009446
299 >>> s = "if hasattr(str, '__bool__'): pass"
300 >>> timeit.timeit(stmt=s, number=100000)
301 0.5829014980008651
302 >>>
303 >>> # attribute is present
Georg Brandl116aa622007-08-15 14:28:22 +0000304 >>> s = """\
305 ... try:
306 ... int.__bool__
307 ... except AttributeError:
308 ... pass
309 ... """
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300310 >>> timeit.timeit(stmt=s, number=100000)
311 0.04215312199994514
312 >>> s = "if hasattr(int, '__bool__'): pass"
313 >>> timeit.timeit(stmt=s, number=100000)
314 0.08588060699912603
315
Georg Brandl116aa622007-08-15 14:28:22 +0000316
317To give the :mod:`timeit` module access to functions you define, you can pass a
Ezio Melottia3ccb232012-09-20 06:13:38 +0300318*setup* parameter which contains an import statement::
Georg Brandl116aa622007-08-15 14:28:22 +0000319
320 def test():
Senthil Kumaran2e015352011-08-06 13:37:04 +0800321 """Stupid test function"""
Collin Winterc79461b2007-09-01 23:34:30 +0000322 L = [i for i in range(100)]
Georg Brandl116aa622007-08-15 14:28:22 +0000323
Senthil Kumaran2e015352011-08-06 13:37:04 +0800324 if __name__ == '__main__':
Ezio Melottid0fe3e52012-10-02 05:35:39 +0300325 import timeit
326 print(timeit.timeit("test()", setup="from __main__ import test"))