Update extract_nums.py

akashsonowal · Dec 5, 2023 · 6c5df90 · 6c5df90
1 parent 9505cc9
commit 6c5df90
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/interview_prep/coding/dsa/extract_nums.py b/interview_prep/coding/dsa/extract_nums.py
@@ -8,16 +8,16 @@ def extract_nums(ip):
         res = []
 
         while i < len(ip):
-        if ip[i].isdigit():
-            l = r = i
-            while l >= 0 and ip[l].isdigit():
-            l -= 1
-            while r < len(ip) and ip[r].isdigit():
-            r += 1
-            res.append(ip[l + 1: r])
-            i = r
-        else:
-            i += 1 
+            if ip[i].isdigit():
+                l = r = i
+                while l >= 0 and ip[l].isdigit():
+                    l -= 1
+                while r < len(ip) and ip[r].isdigit():
+                    r += 1
+                res.append(ip[l + 1: r])
+                i = r
+            else:
+                i += 1 
 
         return res
 
@@ -26,4 +26,4 @@ def extract_nums(ip):
     ip = "avdcfsd3212ad123ad"
     op = ["3212", "123"]
 
-    assert Solution.extract_nums(ip) == op
+    assert Solution.extract_nums(ip) == op