bpo-36461: Improve the code after review comments

Author: mangrisano

Doc/library/timeit.rst

@@ -146,7 +146,7 @@ The module defines three convenience functions and a public class:
       Automatically determine how many times to call :meth:`.timeit`.
 
       This is a convenience function that calls :meth:`.timeit` repeatedly
-      so that the total time is greater than *total_time*, returning the eventual
+      so that the total time is greater than or equal to *total_time*, returning the eventual
       (number of loops, time taken for that number of loops). It calls
       :meth:`.timeit` with increasing numbers from the sequence 1, 2, 5,
       10, 20, 50, ... until the time taken is at least *total_time*.

Lib/test/test_timeit.py

@@ -355,7 +355,7 @@ def test_main_exception_fixed_reps(self):
         self.assert_exc_string(error_stringio.getvalue(), 'ZeroDivisionError')
 
     def autorange(self, seconds_per_increment=1/1024, callback=None,
-            total_time=0.2):
+                    total_time=0.2):
         timer = FakeTimer(seconds_per_increment=seconds_per_increment)
         t = timeit.Timer(stmt=self.fake_stmt, setup=self.fake_setup, timer=timer)
         return t.autorange(callback, total_time)
@@ -382,35 +382,39 @@ def callback(a, b):
             num_loops, time_taken = self.autorange(callback=callback)
         self.assertEqual(num_loops, 500)
         self.assertEqual(time_taken, 500/1024)
-        expected = ('1 0.001\n'
-                    '2 0.002\n'
-                    '5 0.005\n'
-                    '10 0.010\n'
-                    '20 0.020\n'
-                    '50 0.049\n'
-                    '100 0.098\n'
-                    '200 0.195\n'
-                    '500 0.488\n')
+        expected = dedent('''\
+                    1 0.001
+                    2 0.002
+                    5 0.005
+                    10 0.010
+                    20 0.020
+                    50 0.049
+                    100 0.098
+                    200 0.195
+                    500 0.488
+        ''')
         self.assertEqual(s.getvalue(), expected)
 
     def test_autorange_with_callback_and_total_time(self):
         def callback(a, b):
             print("{} {:.3f}".format(a, b))
         with captured_stdout() as s:
             num_loops, time_taken = self.autorange(callback=callback,
-                    total_time=0.6)
+                                                    total_time=0.6)
         self.assertEqual(num_loops, 1000)
         self.assertEqual(time_taken, 1000/1024)
-        expected = ('1 0.001\n'
-                    '2 0.002\n'
-                    '5 0.005\n'
-                    '10 0.010\n'
-                    '20 0.020\n'
-                    '50 0.049\n'
-                    '100 0.098\n'
-                    '200 0.195\n'
-                    '500 0.488\n'
-                    '1000 0.977\n')
+        expected = dedent('''\
+                    1 0.001
+                    2 0.002
+                    5 0.005
+                    10 0.010
+                    20 0.020
+                    50 0.049
+                    100 0.098
+                    200 0.195
+                    500 0.488
+                    1000 0.977
+        ''')
         self.assertEqual(s.getvalue(), expected)
 
 

Lib/timeit.py

@@ -207,7 +207,7 @@ def repeat(self, repeat=default_repeat, number=default_number):
 
     def autorange(self, callback=None, total_time=0.2):
         """Return the number of loops and time taken so that total time
-        is greater than *total_time*.
+        is greater than or equal to *total_time*.
 
         Calls the timeit method with increasing numbers from the sequence
         1, 2, 5, 10, 20, 50, ... until the time taken is at least 0.2