refactor: FloydTriangle

Author: alxkm

src/main/java/com/thealgorithms/others/FloydTriangle.java

@@ -1,22 +1,28 @@
 package com.thealgorithms.others;
 
-import java.util.Scanner;
-
 final class FloydTriangle {
     private FloydTriangle() {
     }
 
-    public static void main(String[] args) {
-        Scanner sc = new Scanner(System.in);
-        System.out.println("Enter the number of rows which you want in your Floyd Triangle: ");
-        int r = sc.nextInt();
-        int n = 0;
-        sc.close();
-        for (int i = 0; i < r; i++) {
+    /**
+     * Generates a Floyd Triangle with the specified number of rows.
+     *
+     * @param rows The number of rows in the triangle.
+     * @return A string representing the Floyd Triangle.
+     */
+    public static String generateFloydTriangle(int rows) {
+        StringBuilder triangle = new StringBuilder();
+        int number = 1;
+
+        for (int i = 0; i < rows; i++) {
             for (int j = 0; j <= i; j++) {
-                System.out.print(++n + " ");
+                triangle.append(number++).append(" ");
+            }
+            if (i < rows - 1) {
+                triangle.append("\n");
             }
-            System.out.println();
         }
+
+        return triangle.toString();
     }
 }

src/test/java/com/thealgorithms/others/FloydTriangleTest.java

@@ -0,0 +1,36 @@
+package com.thealgorithms.others;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+public class FloydTriangleTest {
+
+    @Test
+    public void testGenerateFloydTriangleWithValidInput() {
+        String expectedOutput = "1 \n2 3 \n4 5 6 ";
+        assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(3));
+    }
+
+    @Test
+    public void testGenerateFloydTriangleWithOneRow() {
+        String expectedOutput = "1 ";
+        assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(1));
+    }
+
+    @Test
+    public void testGenerateFloydTriangleWithZeroRows() {
+        assertEquals("", FloydTriangle.generateFloydTriangle(0));
+    }
+
+    @Test
+    public void testGenerateFloydTriangleWithNegativeRows() {
+        assertEquals("", FloydTriangle.generateFloydTriangle(-3));
+    }
+
+    @Test
+    public void testGenerateFloydTriangleWithMultipleRows() {
+        String expectedOutput = "1 \n2 3 \n4 5 6 \n7 8 9 10 \n11 12 13 14 15 ";
+        assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(5));
+    }
+}