diff --git a/test/test_coder.py b/test/test_coder.py
index 80b128dce36dde5a4b7bdf0f033189e4d7ff17a0..8ae4c3a2482dc781d14f71ff8a2ab8a0c30d7619 100644
--- a/test/test_coder.py
+++ b/test/test_coder.py
@@ -7,7 +7,7 @@ from soitool import coder
 
 class CoderTest(unittest.TestCase):
     """Test Coder.py."""
-    
+
     LEN = 2
     COMBINATIONS = len(ascii_uppercase)**LEN
 
@@ -19,7 +19,7 @@ class CoderTest(unittest.TestCase):
     def test_get_code_set(self):
         """Test all codes are unique."""
         codes = coder.get_code_set(self.COMBINATIONS, self.LEN)
-        self.assertAlmostEqual(self.COMBINATIONS, len(codes))
+        self.assertEqual(self.COMBINATIONS, len(codes))
 
         # All possible permutations need to be inn codes
         permutations_li = list(permutations(ascii_uppercase, self.LEN))
@@ -32,6 +32,11 @@ class CoderTest(unittest.TestCase):
 
         self.assertFalse(flag)
 
+    def test_get_code_lenght(self):
+        "Test return correct codelenght."
+        self.assertEqual(self.LEN, coder.get_code_lenght(self.COMBINATIONS))
+        self.assertEqual(3, coder.get_code_lenght(len(ascii_uppercase)**3))
+
 
 if __name__ == '__main__':
     unittest.main()