Improve Base16 Codebase (TheAlgorithms#3534)

* Add doctest and remove input() usage

* Apply suggestions from code review

Co-authored-by: Dhruv Manilawala <[email protected]>

Author: 2 people

Diff for: ciphers/base16.py

@@ -1,13 +1,22 @@
 import base64
 
 
-def main():
-    inp = input("->")
+def encode_to_b16(inp: str) -> bytes:
+    """
+    Encodes a given utf-8 string into base-16.
+    >>> encode_to_b16('Hello World!')
+    b'48656C6C6F20576F726C6421'
+    >>> encode_to_b16('HELLO WORLD!')
+    b'48454C4C4F20574F524C4421'
+    >>> encode_to_b16('')
+    b''
+    """
     encoded = inp.encode("utf-8")  # encoded the input (we need a bytes like object)
     b16encoded = base64.b16encode(encoded)  # b16encoded the encoded string
-    print(b16encoded)
-    print(base64.b16decode(b16encoded).decode("utf-8"))  # decoded it
+    return b16encoded
 
 
 if __name__ == "__main__":
-    main()
+    import doctest
+
+    doctest.testmod()