Fixed unit test for ackermann naive

This commit is contained in:
AdrienCos 2019-06-04 12:12:37 -04:00
parent 3f350c3569
commit c6603e276b
No known key found for this signature in database
GPG key ID: 97AD9AEBCA16D08D

View file

@ -30,6 +30,7 @@ import itertools
from stegano.lsbset import generators
class TestGenerators(unittest.TestCase):
def test_identity(self):
@ -91,17 +92,22 @@ class TestGenerators(unittest.TestCase):
"""Test the Ackermann set.
"""
self.assertEqual(generators.ackermann_naive(3, 1), 13)
self.assertEqual(generators.ackermann_naive(3, 2), 29)
self.assertEqual(generators.ackermann_slow(3, 1), 13)
self.assertEqual(generators.ackermann_slow(3, 2), 29)
def test_ackermann(self):
"""Test the Ackermann set.
"""
with open('./tests/expected-results/ackermann', 'r') as f:
self.assertEqual(generators.ackermann_fast(3, 1), int(f.readline()))
self.assertEqual(generators.ackermann_fast(3, 2), int(f.readline()))
self.assertEqual(generators.ackermann_fast(4, 1), int(f.readline()))
self.assertEqual(generators.ackermann_fast(4, 2), int(f.readline()))
self.assertEqual(generators.ackermann_fast(
3, 1), int(f.readline()))
self.assertEqual(generators.ackermann_fast(
3, 2), int(f.readline()))
self.assertEqual(generators.ackermann_fast(
4, 1), int(f.readline()))
self.assertEqual(generators.ackermann_fast(
4, 2), int(f.readline()))
if __name__ == '__main__':
unittest.main()