def testRemoveUnlikelyPredictionsSingleValues(self):
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.1}, 0.01,
3)
self.assertDictEqual(result, {1: 0.1})
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.001},
0.01, 3)
self.assertDictEqual(result, {1: 0.001})
def testRemoveUnlikelyPredictionsComplex(self):
result = HTMPredictionModel._removeUnlikelyPredictions(
{
1: 0.1,
2: 0.2,
3: 0.3,
4: 0.004
}, 0.01, 3)
self.assertDictEqual(result, {1: 0.1, 2: 0.2, 3: 0.3})
result = HTMPredictionModel._removeUnlikelyPredictions(
{
1: 0.1,
2: 0.2,
3: 0.3,
4: 0.4,
5: 0.005
}, 0.01, 3)
self.assertDictEqual(result, {2: 0.2, 3: 0.3, 4: 0.4})
result = HTMPredictionModel._removeUnlikelyPredictions(
{
1: 0.1,
2: 0.2,
3: 0.3,
4: 0.004,
5: 0.005
}, 0.01, 3)
self.assertDictEqual(result, {1: 0.1, 2: 0.2, 3: 0.3})
def testRemoveUnlikelyPredictionsMaxPredictions(self):
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.1, 2: 0.2, 3: 0.3},
0.01, 3)
self.assertDictEqual(result, {1: 0.1, 2: 0.2, 3: 0.3})
result = HTMPredictionModel._removeUnlikelyPredictions(
{1: 0.1, 2: 0.2, 3: 0.3, 4: 0.4}, 0.01, 3)
self.assertDictEqual(result, {2: 0.2, 3: 0.3, 4: 0.4})
def testRemoveUnlikelyPredictionsLikelihoodThresholds(self):
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.1, 2: 0.001}, 0.01, 3)
self.assertDictEqual(result, {1: 0.1})
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.001, 2: 0.002}, 0.01, 3)
self.assertDictEqual(result, {2: 0.002})
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.002, 2: 0.001}, 0.01, 3)
self.assertDictEqual(result, {1: 0.002})
def testRemoveUnlikelyPredictionsEmpty(self):
result = HTMPredictionModel._removeUnlikelyPredictions({}, 0.01, 3)
self.assertDictEqual(result, {})
def testRemoveUnlikelyPredictionsSingleValues(self):
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.1}, 0.01, 3)
self.assertDictEqual(result, {1: 0.1})
result = HTMPredictionModel._removeUnlikelyPredictions({1: 0.001}, 0.01, 3)
self.assertDictEqual(result, {1: 0.001})
def testRemoveUnlikelyPredictionsEmpty(self):
result = HTMPredictionModel._removeUnlikelyPredictions({}, 0.01, 3)
self.assertDictEqual(result, {})