diff --git a/copycat/temperature.py b/copycat/temperature.py index 5457c97..2a72e64 100644 --- a/copycat/temperature.py +++ b/copycat/temperature.py @@ -1,6 +1,7 @@ import math # Alternate formulas for getAdjustedProbability + def _original(temp, prob): if prob == 0 or prob == 0.5 or temp == 0: return prob diff --git a/tests.py b/tests.py index eccbafd..c7078c6 100644 --- a/tests.py +++ b/tests.py @@ -69,7 +69,7 @@ class TestCopycat(unittest.TestCase): self.fail('No instances of expected key %s were produced! %r != %r' % (k, actual, expected)) def run_testcase(self, initial, modified, target, iterations, expected): - pprint(expected) + #tpprint(expected) actual = Copycat().run(initial, modified, target, iterations) self.assertEqual(sum(a['count'] for a in list(actual.values())), iterations) self.assertProbabilitiesLookRoughlyLike(actual, expected)