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

@ -1,5 +1,5 @@
#!/usr/bin/env python #!/usr/bin/env python
#-*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# Stéganô - Stéganô is a basic Python Steganography module. # Stéganô - Stéganô is a basic Python Steganography module.
# Copyright (C) 2010-2017 Cédric Bonhomme - https://www.cedricbonhomme.org # Copyright (C) 2010-2017 Cédric Bonhomme - https://www.cedricbonhomme.org
@ -30,6 +30,7 @@ import itertools
from stegano.lsbset import generators from stegano.lsbset import generators
class TestGenerators(unittest.TestCase): class TestGenerators(unittest.TestCase):
def test_identity(self): def test_identity(self):
@ -91,17 +92,22 @@ class TestGenerators(unittest.TestCase):
"""Test the Ackermann set. """Test the Ackermann set.
""" """
self.assertEqual(generators.ackermann_naive(3, 1), 13) self.assertEqual(generators.ackermann_slow(3, 1), 13)
self.assertEqual(generators.ackermann_naive(3, 2), 29) self.assertEqual(generators.ackermann_slow(3, 2), 29)
def test_ackermann(self): def test_ackermann(self):
"""Test the Ackermann set. """Test the Ackermann set.
""" """
with open('./tests/expected-results/ackermann', 'r') as f: with open('./tests/expected-results/ackermann', 'r') as f:
self.assertEqual(generators.ackermann_fast(3, 1), int(f.readline())) self.assertEqual(generators.ackermann_fast(
self.assertEqual(generators.ackermann_fast(3, 2), int(f.readline())) 3, 1), int(f.readline()))
self.assertEqual(generators.ackermann_fast(4, 1), int(f.readline())) self.assertEqual(generators.ackermann_fast(
self.assertEqual(generators.ackermann_fast(4, 2), int(f.readline())) 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__': if __name__ == '__main__':
unittest.main() unittest.main()