Flatten

kyu_5/flatten/__init__.py

@@ -0,0 +1 @@
+"""Flatten."""

kyu_5/flatten/flatten.py

@@ -1,12 +1,15 @@
 """
-Solution for -> flatten()
+Solution for -> flatten().
+
 Created by Egor Kostan.
 GitHub: https://github.com/ikostan
 """
 
 
 def flatten(*args) -> list:
     """
+    Flatten function.
+
     The method takes in any number of arguments
     and flattens them into a single array. If any
     of the arguments passed in are an array then
@@ -26,13 +29,16 @@ def flatten(*args) -> list:
 
 def unpack(data, collection: list):
     """
-    Helper method. Unpack data until its not list or a tuple.
+    Unpack helper method.
+
+    Unpack data until its not list or a tuple.
     :param data:
     :param collection:
     :return:
     """
-    if not isinstance(data, list) and not isinstance(data, tuple):
-        collection.append(data)
-    else:
+    if isinstance(data, (list, tuple)):
         for d in data:
             unpack(d, collection)
+    else:
+        collection.append(data)
+

kyu_5/flatten/test_flatten.py

@@ -1,5 +1,6 @@
 """
-Solution for -> flatten()
+Solution for -> flatten().
+
 Created by Egor Kostan.
 GitHub: https://github.com/ikostan
 """
@@ -26,12 +27,12 @@
     name='Source/Kata')
 # pylint: enable-msg=R0801
 class FlattenTestCase(unittest.TestCase):
-    """
-    Testing flatten function
-    """
+    """Testing flatten function."""
 
     def test_flatten(self):
         """
+        Testing flatten function with various test data.
+
         For this exercise you will create a global flatten method.
         The method takes in any number of arguments and flattens
         them into a single array. If any of the arguments passed in