From 05b74acf67b1109fe0a555f179d67d5ab01adfde Mon Sep 17 00:00:00 2001
From: morkolai <nikolai-mork@live.no>
Date: Wed, 1 Apr 2020 13:37:50 +0200
Subject: [PATCH] Cleanup soitool/coder.py

---
 soitool/coder.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/soitool/coder.py b/soitool/coder.py
index 533e259..bbc9e56 100644
--- a/soitool/coder.py
+++ b/soitool/coder.py
@@ -6,8 +6,7 @@ import secrets
 
 
 def get_code(code_length, mode="ascii"):
-    """
-    Generate a single random code.
+    """Generate a single random code.
 
     Parameters
     ----------
@@ -22,6 +21,11 @@ def get_code(code_length, mode="ascii"):
     ------
     code : string
         The code
+
+    Raises
+    ------
+    ValueError
+        If mode param is neither of the accepted
     """
     code = ""
     i = 0
@@ -45,8 +49,7 @@ def get_code(code_length, mode="ascii"):
 
 
 def get_code_set(count, code_length, mode="ascii"):
-    """
-    Generate a set of unique, random codes.
+    """Generate a set of unique, random codes.
 
     Parameters
     ----------
@@ -74,8 +77,7 @@ def get_code_set(count, code_length, mode="ascii"):
 
 
 def get_code_length_needed(number_of_entries):
-    """
-    Get code length needed to support one unique code for each entry.
+    """Get code length needed to support one unique code for each entry.
 
     Parameters
     ----------
-- 
GitLab