def test_running_avg_no_prev(self): fake_test = mock.MagicMock() fake_test.success = 1 fake_test.run_time = None fake_result = { 'start_time': datetime.datetime(1914, 6, 28, 10, 45, 0), 'end_time': datetime.datetime(1914, 6, 28, 10, 45, 50), } fake_values = {} result = shell.running_avg(fake_test, fake_values, fake_result) expected_avg = 50 self.assertEqual(expected_avg, result['run_time'])
def test_running_avg(self): fake_test = mock.MagicMock() fake_test.success = 150 fake_test.run_time = 30.452 fake_result = { 'start_time': datetime.datetime(1914, 6, 28, 10, 45, 0), 'end_time': datetime.datetime(1914, 6, 28, 10, 45, 50), } fake_values = {} result = shell.running_avg(fake_test, fake_values, fake_result) # Let's do some arithmetic expected_avg = ((150 * 30.452) + 50) / 151 # Both values should be 30.581456953642384 self.assertEqual(expected_avg, result['run_time'])