def test_whenConfusionMatrixAreMergedCountsAreAdded(self): rewardClasses = [0,1] confusionMatrix1 = ConfusionMatrix(rewardClasses) confusionMatrix2 = ConfusionMatrix(rewardClasses) realReward = 0 predictedReward = 0 confusionMatrix2.addObservation(realReward, predictedReward) confusionMatrix1.merge(confusionMatrix2) self.assertEqual(1, confusionMatrix1.confusionMatrix[realReward][predictedReward])