diff --git a/soitool/coder.py b/soitool/coder.py index e1a0cca2813d30da4ae8a4fa470c417519a5e26e..2dfde1ad4d81213379f9a0c5de48ac2c59496522 100644 --- a/soitool/coder.py +++ b/soitool/coder.py @@ -4,14 +4,14 @@ import string import secrets -def get_code(lenght): +def get_code(length): """ Generate a single alphabetic random code. Parameters ---------- - lenght : int - The lenght of the code + length : int + The length of the code Return ------ @@ -21,14 +21,14 @@ def get_code(lenght): code = "" i = 0 - while i < lenght: + while i < length: letter = secrets.choice(string.ascii_uppercase) code = code + letter i = i + 1 return code -def get_code_set(size, lenght): +def get_code_set(size, length): """ Generate a set of alphabetic codes. @@ -37,8 +37,8 @@ def get_code_set(size, lenght): size : int Number of codes to be returned - lenght : int - The lenght of the code + length : int + The length of the code Return ------ @@ -48,14 +48,14 @@ def get_code_set(size, lenght): codes = set() while len(codes) < size: - code = get_code(lenght) + code = get_code(length) if code not in codes: codes.add(code) return codes -def get_code_lenght(quantity): +def get_code_length(quantity): """ Get lentgth of code based on number of codewords. @@ -66,12 +66,12 @@ def get_code_lenght(quantity): Return ------ - lenght: int + length: int Lenght of code """ - lenght = 0 - while len(string.ascii_uppercase)**lenght < quantity: - lenght = lenght + 1 + length = 0 + while len(string.ascii_uppercase)**length < quantity: + length = length + 1 - return lenght + return length diff --git a/test/test_coder.py b/test/test_coder.py index 8ae4c3a2482dc781d14f71ff8a2ab8a0c30d7619..a3395a309424521b7e1f5de4c3dd9375d433de74 100644 --- a/test/test_coder.py +++ b/test/test_coder.py @@ -12,7 +12,7 @@ class CoderTest(unittest.TestCase): COMBINATIONS = len(ascii_uppercase)**LEN def test_get_code(self): - """Test code is uppercase alphabetic and correct lenght.""" + """Test code is uppercase alphabetic and correct length.""" code = coder.get_code(3) self.assertRegex(code, '[A-Z]{3}') @@ -34,8 +34,8 @@ class CoderTest(unittest.TestCase): 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)) + self.assertEqual(self.LEN, coder.get_code_length(self.COMBINATIONS)) + self.assertEqual(3, coder.get_code_length(len(ascii_uppercase)**3)) if __name__ == '__main__':