alex/copycat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updates tests

Browse Source
This commit is contained in:
LSaldyt
2017-10-09 14:01:17 -06:00
parent ca38b16e72
commit 6ff43f8d5a
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
copycat/temperature.py
View File

@ -1,6 +1,7 @@
import math import math
# Alternate formulas for getAdjustedProbability # Alternate formulas for getAdjustedProbability
def _original(temp, prob): def _original(temp, prob):
if prob == 0 or prob == 0.5 or temp == 0: if prob == 0 or prob == 0.5 or temp == 0:
return prob return prob

2
tests.py
View File

@ -69,7 +69,7 @@ class TestCopycat(unittest.TestCase):
self.fail('No instances of expected key %s were produced! %r != %r' % (k, actual, expected)) self.fail('No instances of expected key %s were produced! %r != %r' % (k, actual, expected))
def run_testcase(self, initial, modified, target, iterations, expected): def run_testcase(self, initial, modified, target, iterations, expected):
pprint(expected) #tpprint(expected)
actual = Copycat().run(initial, modified, target, iterations) actual = Copycat().run(initial, modified, target, iterations)
self.assertEqual(sum(a['count'] for a in list(actual.values())), iterations) self.assertEqual(sum(a['count'] for a in list(actual.values())), iterations)
self.assertProbabilitiesLookRoughlyLike(actual, expected) self.assertProbabilitiesLookRoughlyLike(actual, expected)