Solution and Test for Issue llipio#116

Himanshu1495 · Sep 19, 2017 · a4d21ec · a4d21ec
1 parent 8009870
commit a4d21ec
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/solutions/sol116.py b/solutions/sol116.py
@@ -0,0 +1,15 @@
+#Himanshu Nachane (@Himanshu1495)
+'''input: [1, 2, [2, [5, 3, 4]] ]
+output: [1,2, 3, 5, 3, 4]
+
+the output order does not matter, just so long as the array is flattened
+'''
+
+def flatten_array(arr):
+    for i in arr:
+        if len(str(i)) == 1:
+            store.append(i)
+        else:
+            flatten_array(i)
+    return store
+store = []
diff --git a/test/116.py b/test/116.py
@@ -0,0 +1,10 @@
+import sys
+sys.path.insert(0,'../solutions/')
+from sol116 import flatten_array
+import unittest
+
+class MyTest(unittest.TestCase):
+    def test1(self):
+        self.assertEqual(flatten_array([1,2,[3,4,5],[6,[7,8]],9]),[1,2,3,4,5,6,7,8,9])
+
+unittest.main()