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
#-*- coding: utf-8 -*-
# -*- coding: utf-8 -*-
# Stéganô - Stéganô is a basic Python Steganography module.
# Copyright (C) 2010-2017 Cédric Bonhomme - https://www.cedricbonhomme.org
@ -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()