Lines Matching refs:self
24 def __init__(self, seconds_per_increment=1.0):
25 self.count = 0
26 self.setup_calls = 0
27 self.seconds_per_increment=seconds_per_increment
28 timeit._fake_timer = self
30 def __call__(self):
31 return self.BASE_TIME + self.count * self.seconds_per_increment
33 def inc(self):
34 self.count += 1
36 def setup(self):
37 self.setup_calls += 1
39 def wrap_timer(self, timer):
40 """Records 'timer' and returns self as callable timer."""
41 self.saved_timer = timer
42 return self
46 def tearDown(self):
52 def test_reindent_empty(self):
53 self.assertEqual(timeit.reindent("", 0), "")
54 self.assertEqual(timeit.reindent("", 4), "")
56 def test_reindent_single(self):
57 self.assertEqual(timeit.reindent("pass", 0), "pass")
58 self.assertEqual(timeit.reindent("pass", 4), "pass")
60 def test_reindent_multi_empty(self):
61 self.assertEqual(timeit.reindent("\n\n", 0), "\n\n")
62 self.assertEqual(timeit.reindent("\n\n", 4), "\n \n ")
64 def test_reindent_multi(self):
65 self.assertEqual(timeit.reindent(
68 self.assertEqual(timeit.reindent(
72 def test_timer_invalid_stmt(self):
73 self.assertRaises(ValueError, timeit.Timer, stmt=None)
74 self.assertRaises(SyntaxError, timeit.Timer, stmt='return')
75 self.assertRaises(SyntaxError, timeit.Timer, stmt='yield')
76 self.assertRaises(SyntaxError, timeit.Timer, stmt='yield from ()')
77 self.assertRaises(SyntaxError, timeit.Timer, stmt='break')
78 self.assertRaises(SyntaxError, timeit.Timer, stmt='continue')
79 self.assertRaises(SyntaxError, timeit.Timer, stmt='from timeit import *')
80 self.assertRaises(SyntaxError, timeit.Timer, stmt=' pass')
81 self.assertRaises(SyntaxError, timeit.Timer,
84 def test_timer_invalid_setup(self):
85 self.assertRaises(ValueError, timeit.Timer, setup=None)
86 self.assertRaises(SyntaxError, timeit.Timer, setup='return')
87 self.assertRaises(SyntaxError, timeit.Timer, setup='yield')
88 self.assertRaises(SyntaxError, timeit.Timer, setup='yield from ()')
89 self.assertRaises(SyntaxError, timeit.Timer, setup='break')
90 self.assertRaises(SyntaxError, timeit.Timer, setup='continue')
91 self.assertRaises(SyntaxError, timeit.Timer, setup='from timeit import *')
92 self.assertRaises(SyntaxError, timeit.Timer, setup=' pass')
94 def test_timer_empty_stmt(self):
102 def fake_callable_setup(self):
103 self.fake_timer.setup()
105 def fake_callable_stmt(self):
106 self.fake_timer.inc()
108 def timeit(self, stmt, setup, number=None, globals=None):
109 self.fake_timer = FakeTimer()
110 t = timeit.Timer(stmt=stmt, setup=setup, timer=self.fake_timer,
118 self.assertEqual(self.fake_timer.setup_calls, 1)
119 self.assertEqual(self.fake_timer.count, number)
120 self.assertEqual(delta_time, number)
123 #def test_timeit_default_iters(self):
124 # self.timeit(self.fake_stmt, self.fake_setup)
126 def test_timeit_zero_iters(self):
127 self.timeit(self.fake_stmt, self.fake_setup, number=0)
129 def test_timeit_few_iters(self):
130 self.timeit(self.fake_stmt, self.fake_setup, number=3)
132 def test_timeit_callable_stmt(self):
133 self.timeit(self.fake_callable_stmt, self.fake_setup, number=3)
135 def test_timeit_callable_setup(self):
136 self.timeit(self.fake_stmt, self.fake_callable_setup, number=3)
138 def test_timeit_callable_stmt_and_setup(self):
139 self.timeit(self.fake_callable_stmt,
140 self.fake_callable_setup, number=3)
143 #def test_timeit_function(self):
144 # delta_time = timeit.timeit(self.fake_stmt, self.fake_setup,
146 # self.assertEqual(delta_time, DEFAULT_NUMBER)
148 def test_timeit_function_zero_iters(self):
149 delta_time = timeit.timeit(self.fake_stmt, self.fake_setup, number=0,
151 self.assertEqual(delta_time, 0)
153 def test_timeit_globals_args(self):
157 self.assertRaises(NameError, t.timeit, number=3)
164 def repeat(self, stmt, setup, repeat=None, number=None):
165 self.fake_timer = FakeTimer()
166 t = timeit.Timer(stmt=stmt, setup=setup, timer=self.fake_timer)
177 self.assertEqual(self.fake_timer.setup_calls, repeat)
178 self.assertEqual(self.fake_timer.count, repeat * number)
179 self.assertEqual(delta_times, repeat * [float(number)])
182 #def test_repeat_default(self):
183 # self.repeat(self.fake_stmt, self.fake_setup)
185 def test_repeat_zero_reps(self):
186 self.repeat(self.fake_stmt, self.fake_setup, repeat=0)
188 def test_repeat_zero_iters(self):
189 self.repeat(self.fake_stmt, self.fake_setup, number=0)
191 def test_repeat_few_reps_and_iters(self):
192 self.repeat(self.fake_stmt, self.fake_setup, repeat=3, number=5)
194 def test_repeat_callable_stmt(self):
195 self.repeat(self.fake_callable_stmt, self.fake_setup,
198 def test_repeat_callable_setup(self):
199 self.repeat(self.fake_stmt, self.fake_callable_setup,
202 def test_repeat_callable_stmt_and_setup(self):
203 self.repeat(self.fake_callable_stmt, self.fake_callable_setup,
207 #def test_repeat_function(self):
208 # delta_times = timeit.repeat(self.fake_stmt, self.fake_setup,
210 # self.assertEqual(delta_times, DEFAULT_REPEAT * [float(DEFAULT_NUMBER)])
212 def test_repeat_function_zero_reps(self):
213 delta_times = timeit.repeat(self.fake_stmt, self.fake_setup, repeat=0,
215 self.assertEqual(delta_times, [])
217 def test_repeat_function_zero_iters(self):
218 delta_times = timeit.repeat(self.fake_stmt, self.fake_setup, number=0,
220 self.assertEqual(delta_times, DEFAULT_REPEAT * [0.0])
222 def assert_exc_string(self, exc_string, expected_exc_name):
224 self.assertGreater(len(exc_lines), 2)
225 self.assertTrue(exc_lines[0].startswith('Traceback'))
226 self.assertTrue(exc_lines[-1].startswith(expected_exc_name))
228 def test_print_exc(self):
235 self.assert_exc_string(s.getvalue(), 'ZeroDivisionError')
239 def run_main(self, seconds_per_increment=1.0, switches=None, timer=None):
246 args.append(self.fake_stmt)
254 def test_main_bad_switch(self):
255 s = self.run_main(switches=['--bad-switch'])
256 self.assertEqual(s, dedent("""\
261 def test_main_seconds(self):
262 s = self.run_main(seconds_per_increment=5.5)
263 self.assertEqual(s, "1 loop, best of 5: 5.5 sec per loop\n")
265 def test_main_milliseconds(self):
266 s = self.run_main(seconds_per_increment=0.0055)
267 self.assertEqual(s, "50 loops, best of 5: 5.5 msec per loop\n")
269 def test_main_microseconds(self):
270 s = self.run_main(seconds_per_increment=0.0000025, switches=['-n100'])
271 self.assertEqual(s, "100 loops, best of 5: 2.5 usec per loop\n")
273 def test_main_fixed_iters(self):
274 s = self.run_main(seconds_per_increment=2.0, switches=['-n35'])
275 self.assertEqual(s, "35 loops, best of 5: 2 sec per loop\n")
277 def test_main_setup(self):
278 s = self.run_main(seconds_per_increment=2.0,
280 self.assertEqual(s, "CustomSetup\n" * DEFAULT_REPEAT +
283 def test_main_multiple_setups(self):
284 s = self.run_main(seconds_per_increment=2.0,
286 self.assertEqual(s, "CustomSetup\n" * DEFAULT_REPEAT +
289 def test_main_fixed_reps(self):
290 s = self.run_main(seconds_per_increment=60.0, switches=['-r9'])
291 self.assertEqual(s, "1 loop, best of 9: 60 sec per loop\n")
293 def test_main_negative_reps(self):
294 s = self.run_main(seconds_per_increment=60.0, switches=['-r-5'])
295 self.assertEqual(s, "1 loop, best of 1: 60 sec per loop\n")
298 def test_main_help(self):
299 s = self.run_main(switches=['-h'])
302 self.assertEqual(s, timeit.__doc__ + ' ')
304 def test_main_verbose(self):
305 s = self.run_main(switches=['-v'])
306 self.assertEqual(s, dedent("""\
314 def test_main_very_verbose(self):
315 s = self.run_main(seconds_per_increment=0.000_030, switches=['-vv'])
316 self.assertEqual(s, dedent("""\
336 def test_main_with_time_unit(self):
337 unit_sec = self.run_main(seconds_per_increment=0.003,
339 self.assertEqual(unit_sec,
341 unit_msec = self.run_main(seconds_per_increment=0.003,
343 self.assertEqual(unit_msec,
345 unit_usec = self.run_main(seconds_per_increment=0.003,
347 self.assertEqual(unit_usec,
351 invalid = self.run_main(seconds_per_increment=0.003,
353 self.assertEqual(error_stringio.getvalue(),
356 def test_main_exception(self):
358 s = self.run_main(switches=['1/0'])
359 self.assert_exc_string(error_stringio.getvalue(), 'ZeroDivisionError')
361 def test_main_exception_fixed_reps(self):
363 s = self.run_main(switches=['-n1', '1/0'])
364 self.assert_exc_string(error_stringio.getvalue(), 'ZeroDivisionError')
366 def autorange(self, seconds_per_increment=1/1024, callback=None):
368 t = timeit.Timer(stmt=self.fake_stmt, setup=self.fake_setup, timer=timer)
371 def test_autorange(self):
372 num_loops, time_taken = self.autorange()
373 self.assertEqual(num_loops, 500)
374 self.assertEqual(time_taken, 500/1024)
376 def test_autorange_second(self):
377 num_loops, time_taken = self.autorange(seconds_per_increment=1.0)
378 self.assertEqual(num_loops, 1)
379 self.assertEqual(time_taken, 1.0)
381 def test_autorange_with_callback(self):
385 num_loops, time_taken = self.autorange(callback=callback)
386 self.assertEqual(num_loops, 500)
387 self.assertEqual(time_taken, 500/1024)
397 self.assertEqual(s.getvalue(), expected)