From 82b59f73dbbf32985560cc7b1c640c9f039c6f30 Mon Sep 17 00:00:00 2001
From: Flavien <flavien@derniercri.io>
Date: Fri, 11 Nov 2022 14:19:34 +0100
Subject: [PATCH] Add test with mersenne generator

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

diff --git a/tests/test_lsb.py b/tests/test_lsb.py
index 1145075..a50620c 100644
--- a/tests/test_lsb.py
+++ b/tests/test_lsb.py
@@ -90,6 +90,20 @@ class TestLSB(unittest.TestCase):
 
             self.assertEqual(message, clear_message)
 
+    def test_hide_and_reveal_with_mersenne(self):
+        messages_to_hide = ["f"]
+        for message in messages_to_hide:
+            secret = lsb.hide(
+                "./tests/sample-files/Lenna.png",
+                message,
+                generators.mersenne(),
+            )
+            secret.save("./image.png")
+
+            clear_message = lsb.reveal("./image.png", generators.mersenne())
+
+            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: