From 54938159d4612512bfb18f5ba4e675c7eec751b6 Mon Sep 17 00:00:00 2001
From: Flavien <flavien@derniercri.io>
Date: Fri, 11 Nov 2022 14:07:40 +0100
Subject: [PATCH] Add test with ackermann_naive generator

---
 tests/test_lsb.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/test_lsb.py b/tests/test_lsb.py
index ca3fd47..1145075 100644
--- a/tests/test_lsb.py
+++ b/tests/test_lsb.py
@@ -76,6 +76,20 @@ class TestLSB(unittest.TestCase):
 
             self.assertEqual(message, clear_message)
 
+    def test_hide_and_reveal_with_ackermann_naive(self):
+        messages_to_hide = ["foo"]
+        for message in messages_to_hide:
+            secret = lsb.hide(
+                "./tests/sample-files/Lenna.png",
+                message,
+                generators.ackermann_naive(m=2),
+            )
+            secret.save("./image.png")
+
+            clear_message = lsb.reveal("./image.png", generators.ackermann_naive(m=2))
+
+            self.assertEqual(message, clear_message)
+
     def test_hide_and_reveal_with_shi_tomashi(self):
         messages_to_hide = ["foo bar"]
         for message in messages_to_hide: