def test_whenMoveRecordedPredictedRewardIsRemmembered(self): possibleRewards = [0,1] performanceRecord = PerformanceRecord(possibleRewards) currentReward = 1 nextStatePredictedReward = 2 performanceRecord.recordMove(currentReward, nextStatePredictedReward) self.assertEqual(nextStatePredictedReward, performanceRecord.predictedReward)
def test_whenMoveRecordedCurrentRewardIsAddedToAccumulatedReward(self): possibleRewards = [0,1] performanceRecord = PerformanceRecord(possibleRewards) currentReward = 1 nextStatePredictedReward = 2 performanceRecord.recordMove(currentReward, nextStatePredictedReward) self.assertEqual(1, performanceRecord.accumulatedReward)
def test_whenMoveRecordedNoVisitedStatesIncreases(self): possibleRewards = [0,1] performanceRecord = PerformanceRecord(possibleRewards) currentReward = 1 nextStatePredictedReward = 2 performanceRecord.recordMove(currentReward, nextStatePredictedReward) self.assertEqual(0, performanceRecord.noVisitedStates)
def test_whenIterationIsNewUserRecordConfusionMatrix(self):
possibleRewards = [0,1]
performanceRecord = PerformanceRecord(possibleRewards)
performanceRecord.recordMove(1, 2)
performanceResult = PerformanceResult()
iteration = 1
performanceResult.addRecord(iteration, performanceRecord)
self.assertEqual(performanceRecord.confusionMatrix, performanceResult.results[iteration]["ConfusionMatrix"])
def test_whenRecordIsAddedAccumulatedRewardIsAppended(self): possibleRewards = [0,2,1] performanceRecord1 = PerformanceRecord(possibleRewards) performanceRecord1.recordMove(1, 1) performanceRecord2 = PerformanceRecord(possibleRewards) performanceRecord2.recordMove(2, 1) iteration = 1 performanceResult = PerformanceResult() performanceResult.addRecord(iteration, performanceRecord1) performanceResult.addRecord(iteration, performanceRecord2) self.assertEqual(1, performanceResult.results[iteration]["AccumulatedReward"][0]) self.assertEqual(2, performanceResult.results[iteration]["AccumulatedReward"][1])
def test_whenIterationExistsMergeConfusionMatrices(self): possibleRewards = [0,1] performanceRecord1 = PerformanceRecord(possibleRewards) performanceRecord1.recordMove(0, 0) performanceRecord2 = PerformanceRecord(possibleRewards) performanceRecord2.recordMove(1, 1) iteration = 1 performanceResult = PerformanceResult() performanceResult.addRecord(iteration, performanceRecord1) performanceResult.addRecord(iteration, performanceRecord2) self.assertEqual(1, performanceResult.results[iteration]["ConfusionMatrix"].confusionMatrix[1][0]) self.assertEqual(1, performanceResult.results[iteration]["ConfusionMatrix"].confusionMatrix[0][0])
