adding algo

Author: ivanpenaloza

src/my_project/interviews/top_150_questions_round_15/sqrtx.py

@@ -0,0 +1,20 @@
+from typing import List, Union, Collection, Mapping, Optional
+from abc import ABC, abstractmethod
+
+class Solution:
+    def mySqrt(self, x: int) -> int:
+
+        left, right = 0, x 
+
+        while left <= right: 
+            
+            mid = (left + right)//2
+
+            if mid ** 2 > x:
+                right  = mid - 1
+            elif mid ** 2 < x:
+                left = mid + 1
+            else: 
+                return mid 
+            
+        return min(left, right)

tests/test_150_questions_round_14/test_sqrtx_round_14.py

@@ -2,20 +2,17 @@
 from src.my_project.interviews.top_150_questions_round_14\
 .sqrtx import Solution
 
-class Solution:
-    def mySqrt(self, x: int) -> int:
 
-        left, right = 0, x 
+class SqrtxTestCase(unittest.TestCase):
 
-        while left <= right:
+    def test_even_sqrtx(self):
+        solution = Solution()
+        output = solution.mySqrt(x=4)
+        target = 2 
+        self.assertEqual(output, target)
 
-            mid = (left + right)//2
-
-            if mid ** 2 > x: 
-                right = mid - 1
-            elif mid ** 2 < x: 
-                left = mid + 1 
-            else: 
-                return mid 
-        
-        return min(left, right)
+    def test_odd_sqrtx(self):
+        solution = Solution()
+        output = solution.mySqrt(x=8)
+        target = 2 
+        self.assertEqual(output, target)

tests/test_150_questions_round_15/test_sqrtx_round_15.py

@@ -0,0 +1,18 @@
+import unittest
+from src.my_project.interviews.top_150_questions_round_15\
+.sqrtx import Solution
+
+
+class SqrtxTestCase(unittest.TestCase):
+
+    def test_even_sqrtx(self):
+        solution = Solution()
+        output = solution.mySqrt(x=4)
+        target = 2 
+        self.assertEqual(output, target)
+
+    def test_odd_sqrtx(self):
+        solution = Solution()
+        output = solution.mySqrt(x=8)
+        target = 2 
+        self.assertEqual(output, target)