def calculateCollisionTimesAverage(simulation):
accumulatedTimes = [step.time for step in simulation.steps]
deltaTimes = calculateDeltas(accumulatedTimes)
return average(deltaTimes)
def testUpperBoundAverage(self): dataset = [MAX_VALUE, MAX_VALUE, MAX_VALUE] average = calculator.average(dataset) self.assertEqual(average, MAX_VALUE)
def testNormalAverage(self): dataset = [1, 3, 5] average = calculator.average(dataset) self.assertEqual(average, 3)
def testInvalidAverage(self):
dataset = [23, 45, "ok"]
with self.assertRaises(Exception):
average = calculator.average(dataset)