modified the file

Author: vivek2004-sec

strings/Secret_language.py

@@ -6,55 +6,69 @@
 
 def random_chars() -> str:
     """
-    Adds random 3 charaters to the string.
+    Generate a random string of 3 ASCII letters.
 
+    >>> import random
+    >>> random.seed(42)
+    >>> random_chars()
+    'ZoX'
     """
-    return "".join(random.choices(string.ascii_letters, k=3))
+    return ''.join(random.choices(string.ascii_letters, k=3))
 
 
 def random_digits() -> str:
     """
-    Adds 3 random digits to the string.
-    """
+    Generate a random string of 3 digits.
 
-    return "".join(random.choices(string.digits, k=3))
+    >>> import random
+    >>> random.seed(42)
+    >>> random_digits()
+    '638'
+    """
+    return ''.join(random.choices(string.digits, k=3))
 
 
-<<<<<<< HEAD
 def encode(code: str) -> str:
-    '''
-=======
-def encode(code) -> str:
     """
->>>>>>> cb237ed2095a60e62d73afd8b17c1207cc1fb4f9
-    Encodes the code by shifting the first character to the end of the original string,
-    and adding the 3 random_characters + 3 random-digits + original string(code) +  3 random-digits + 3 random_characters.
-    unless the length of the code exceeds 3 or equals to it."""
+    Encode a string by shifting the first character to the end and
+    wrapping it with random padding of 3 letters and 3 digits on each side.
+
+    Reference: https://en.wikipedia.org/wiki/Caesar_cipher
 
+    >>> import random
+    >>> random.seed(42)
+    >>> encode('hello')
+    'ZoX638elloh415mJu'
+
+    >>> import random
+    >>> random.seed(42)
+    >>> encode('hi')
+    'ZoX638ih415mJu'
+    """
     if len(code) >= 3:
         code = code[1:] + code[0]
-        code = (
-            random_chars() + random_digits() + code + random_digits() + random_chars()
-        )
+        code = random_chars() + random_digits() + code + random_digits() + random_chars()
     else:
         code = code[::-1]
-        code = (
-            random_chars() + random_digits() + code + random_digits() + random_chars()
-        )
+        code = random_chars() + random_digits() + code + random_digits() + random_chars()
     return code
 
-<<<<<<< HEAD
-def decode(code: str) -> str:
-    '''
-=======
 
-def decode(code) -> str:
+def decode(code: str) -> str:
     """
->>>>>>> cb237ed2095a60e62d73afd8b17c1207cc1fb4f9
-    decodes the encoded string by removing the randomly added characters and reversing the shift
-
+    Decode an encoded string by stripping the random padding and
+    reversing the character shift.
+
+    >>> import random
+    >>> random.seed(42)
+    >>> decode(encode('hello'))
+    'hello'
+
+    >>> import random
+    >>> random.seed(42)
+    >>> decode(encode('hi'))
+    'hi'
     """
-
     code = code[6:-6]
     if len(code) >= 3:
         code = code[-1] + code[:-1]
@@ -69,4 +83,4 @@ def decode(code) -> str:
     decoded = decode(encoded)
     print(f"Original → {code}")
     print(f"Encoded  → {encoded}")
-    print(f"Decoded  → {decoded}")
+    print(f"Decoded  → {decoded}")