fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 17:18:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed unit test for ackermann naive

Browse source
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
1 changed files with 24 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
tests/test_generators.py
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,60 +30,61 @@ 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):
"""Test the identity generator. """Test the identity generator.
""" """
self.assertEqual(tuple(itertools.islice(generators.identity(), 15)), self.assertEqual(tuple(itertools.islice(generators.identity(), 15)),
(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14)) (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14))
def test_fibonacci(self): def test_fibonacci(self):
"""Test the Fibonacci generator. """Test the Fibonacci generator.
""" """
self.assertEqual(tuple(itertools.islice(generators.fibonacci(), 20)), self.assertEqual(tuple(itertools.islice(generators.fibonacci(), 20)),
(1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610, (1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610,
987, 1597, 2584, 4181, 6765, 10946)) 987, 1597, 2584, 4181, 6765, 10946))
def test_eratosthenes(self): def test_eratosthenes(self):
"""Test the Eratosthenes sieve. """Test the Eratosthenes sieve.
""" """
with open('./tests/expected-results/eratosthenes', 'r') as f: with open('./tests/expected-results/eratosthenes', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.eratosthenes(), 168)), self.assertEqual(tuple(itertools.islice(generators.eratosthenes(), 168)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
def test_composite(self): def test_composite(self):
"""Test the composite sieve. """Test the composite sieve.
""" """
with open('./tests/expected-results/composite', 'r') as f: with open('./tests/expected-results/composite', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.composite(), 114)), self.assertEqual(tuple(itertools.islice(generators.composite(), 114)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
def test_fermat(self): def test_fermat(self):
"""Test the Fermat generator. """Test the Fermat generator.
""" """
with open('./tests/expected-results/fermat', 'r') as f: with open('./tests/expected-results/fermat', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.fermat(), 9)), self.assertEqual(tuple(itertools.islice(generators.fermat(), 9)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
def test_triangular_numbers(self): def test_triangular_numbers(self):
"""Test the Triangular numbers generator. """Test the Triangular numbers generator.
""" """
with open('./tests/expected-results/triangular_numbers', 'r') as f: with open('./tests/expected-results/triangular_numbers', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.triangular_numbers(), 54)), self.assertEqual(tuple(itertools.islice(generators.triangular_numbers(), 54)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
def test_mersenne(self): def test_mersenne(self):
"""Test the Mersenne generator. """Test the Mersenne generator.
""" """
with open('./tests/expected-results/mersenne', 'r') as f: with open('./tests/expected-results/mersenne', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.mersenne(), 20)), self.assertEqual(tuple(itertools.islice(generators.mersenne(), 20)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
def test_carmichael(self): def test_carmichael(self):
"""Test the Carmichael generator. """Test the Carmichael generator.
""" """
with open('./tests/expected-results/carmichael', 'r') as f: with open('./tests/expected-results/carmichael', 'r') as f:
self.assertEqual(tuple(itertools.islice(generators.carmichael(), 33)), self.assertEqual(tuple(itertools.islice(generators.carmichael(), 33)),
tuple(int(line) for line in f)) tuple(int(line) for line in f))
@ -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()