From 2a2a66fd2f09d56b04e054632bc6ecea5b554b33 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stefan=20Fr=C3=B6se?= <stefan.froese@udo.edu>
Date: Thu, 13 Jul 2023 13:37:59 +0200
Subject: [PATCH] import timeit

---
 examples/tutorials/ctapipe_overview.py | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/examples/tutorials/ctapipe_overview.py b/examples/tutorials/ctapipe_overview.py
index 1b6c1dc4a02..e96cea70f45 100644
--- a/examples/tutorials/ctapipe_overview.py
+++ b/examples/tutorials/ctapipe_overview.py
@@ -12,6 +12,7 @@
 
 
 import tempfile
+import timeit
 from copy import deepcopy
 
 import astropy.units as u
@@ -610,10 +611,7 @@ def num_islands_python(camera, clean):
 disp.add_colorbar()
 
 ######################################################################
-#     .. code-block:: bash
-#
-#       %%timeit
-#       num_islands_python(geometry, clean)
+timeit.timeit(lambda: num_islands_python(geometry, clean), number=1000) / 1000
 
 
 ######################################################################
@@ -640,11 +638,7 @@ def num_islands_scipy(geometry, clean):
 disp.add_colorbar()
 
 ######################################################################
-#     .. code-block:: bash
-#
-#       %%timeit
-#       num_islands_scipy(geometry, clean)
-
+timeit.timeit(lambda: num_islands_scipy(geometry, clean), number=10000) / 10000
 
 ######################################################################
 # **A lot less code, and a factor 3 speed improvement**
@@ -656,7 +650,4 @@ def num_islands_scipy(geometry, clean):
 #
 
 ######################################################################
-#     .. code-block:: bash
-#
-#       %%timeit
-#       number_of_islands(geometry, clean)
+timeit.timeit(lambda: number_of_islands(geometry, clean), number=100000) / 100000