test: Add some more tests for checking square root.

Vishwajeet-29-pro · Sep 23, 2024 · 2ffe805 · 2ffe805
1 parent cc2f5aa
commit 2ffe805
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/JavaDsaWithTest/src/test/java/org/practice/dsa/leet_code/easy/SquareRootTest.java b/JavaDsaWithTest/src/test/java/org/practice/dsa/leet_code/easy/SquareRootTest.java
@@ -14,5 +14,20 @@ public void testSquareRootForAbsoluteValue() {
         int result = squareRoot.mySqrt(input);
         assertEquals(expected, result);
     }
+
+    @Test
+    public void testSquareRootForEight() {
+        int input = 8;
+        int expected = 2;
+        assertEquals(expected, squareRoot.mySqrt(input));
+    }
+
+    @Test
+    public void testMultipleSquareRoots() {
+        assertEquals(3, squareRoot.mySqrt(9));
+        assertEquals(3, squareRoot.mySqrt(10));
+        assertEquals(6, squareRoot.mySqrt(36));
+        assertEquals(6, squareRoot.mySqrt(37));
+    }
 }