Fix

Author: Hardvan

src/test/java/com/thealgorithms/misc/MatrixTransposeTest.java

@@ -3,56 +3,31 @@
 import static org.junit.jupiter.api.Assertions.assertArrayEquals;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
-import org.junit.jupiter.api.Test;
+import java.util.stream.Stream;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
 
 public class MatrixTransposeTest {
 
-    @Test
-    public void testTransposeSquareMatrix() {
-        int[][] matrix = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
-
-        int[][] expected = {{1, 4, 7}, {2, 5, 8}, {3, 6, 9}};
-
-        assertArrayEquals(expected, MatrixTranspose.transpose(matrix), "Transpose of the square matrix is incorrect.");
+    private static Stream<Arguments> provideValidMatrixTestCases() {
+        return Stream.of(Arguments.of(new int[][] {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}, new int[][] {{1, 4, 7}, {2, 5, 8}, {3, 6, 9}}, "Transpose of square matrix"), Arguments.of(new int[][] {{1, 2}, {3, 4}, {5, 6}}, new int[][] {{1, 3, 5}, {2, 4, 6}}, "Transpose of rectangular matrix"),
+            Arguments.of(new int[][] {{1, 2, 3}}, new int[][] {{1}, {2}, {3}}, "Transpose of single-row matrix"), Arguments.of(new int[][] {{1}, {2}, {3}}, new int[][] {{1, 2, 3}}, "Transpose of single-column matrix"));
     }
 
-    @Test
-    public void testTransposeRectangularMatrix() {
-        int[][] matrix = {{1, 2}, {3, 4}, {5, 6}};
-
-        int[][] expected = {{1, 3, 5}, {2, 4, 6}};
-
-        assertArrayEquals(expected, MatrixTranspose.transpose(matrix), "Transpose of the rectangular matrix is incorrect.");
+    private static Stream<Arguments> provideInvalidMatrixTestCases() {
+        return Stream.of(Arguments.of(new int[0][0], "Empty matrix should throw IllegalArgumentException"), Arguments.of(null, "Null matrix should throw IllegalArgumentException"));
     }
 
-    @Test
-    public void testTransposeSingleRowMatrix() {
-        int[][] matrix = {{1, 2, 3}};
-
-        int[][] expected = {{1}, {2}, {3}};
-
-        assertArrayEquals(expected, MatrixTranspose.transpose(matrix), "Transpose of the single-row matrix is incorrect.");
-    }
-
-    @Test
-    public void testTransposeSingleColumnMatrix() {
-        int[][] matrix = {{1}, {2}, {3}};
-
-        int[][] expected = {{1, 2, 3}};
-
-        assertArrayEquals(expected, MatrixTranspose.transpose(matrix), "Transpose of the single-column matrix is incorrect.");
-    }
-
-    @Test
-    public void testTransposeEmptyMatrix() {
-        int[][] matrix = new int[0][0];
-
-        assertThrows(IllegalArgumentException.class, () -> MatrixTranspose.transpose(matrix), "Expected IllegalArgumentException for empty matrix.");
+    @ParameterizedTest(name = "Test case {index}: {2}")
+    @MethodSource("provideValidMatrixTestCases")
+    void testValidMatrixTranspose(int[][] input, int[][] expected, String description) {
+        assertArrayEquals(expected, MatrixTranspose.transpose(input), description);
     }
 
-    @Test
-    public void testTransposeNullMatrix() {
-        int[][] matrix = null;
-        assertThrows(IllegalArgumentException.class, () -> MatrixTranspose.transpose(matrix), "Expected IllegalArgumentException for null matrix.");
+    @ParameterizedTest(name = "Test case {index}: {1}")
+    @MethodSource("provideInvalidMatrixTestCases")
+    void testInvalidMatrixTranspose(int[][] input, String description) {
+        assertThrows(IllegalArgumentException.class, () -> MatrixTranspose.transpose(input), description);
     }
 }