def test_whenNoPredictionIsCorrectAccuracyIs0(self): rewardClasses = [0,1] confusionMatrix = ConfusionMatrix(rewardClasses) realReward = 0 predictedReward = 1 confusionMatrix.addObservation(realReward, predictedReward) self.assertEqual(0, confusionMatrix.accuracy())
def test_whenAllPredictionsAreCorrectAccuracyIs100(self): rewardClasses = [0,1] confusionMatrix = ConfusionMatrix(rewardClasses) realReward = 0 predictedReward = 0 confusionMatrix.addObservation(realReward, predictedReward) self.assertEqual(100, confusionMatrix.accuracy())