def test_bench_jobs_with_run_seconds(self): initial_handler = lambda s, f: 17 signal.signal(signal.SIGALRM, initial_handler) self.scenario_dict['operation_count'] = 1 self.scenario_dict['run_seconds'] = 1 self.write_scenario_file() scenario = Scenario(self.stub_scenario_file) start_time = time.time() jobs = list(scenario.bench_jobs()) delta_t = time.time() - start_time # Count should be greater than 1, for sure... assert_greater(len(jobs), 1) # +/- 10ms seems good: assert_almost_equal(delta_t, scenario.run_seconds, delta=0.01) restored_handler = signal.signal(signal.SIGALRM, signal.SIG_DFL) assert_equal(restored_handler, initial_handler)