FloydTriangleTest.java

packagecom.thealgorithms.others;

importstaticjava.util.Collections.emptyList;
importstaticjava.util.Collections.singletonList;
importstaticorg.junit.jupiter.api.Assertions.assertEquals;

importjava.util.Arrays;
importjava.util.List;
importorg.junit.jupiter.api.Test;

publicclassFloydTriangleTest {

@Test
publicvoidtestGenerateFloydTriangleWithValidInput() {
List<List<Integer>> expectedOutput = Arrays.asList(singletonList(1), Arrays.asList(2, 3), Arrays.asList(4, 5, 6));
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(3));
 }

@Test
publicvoidtestGenerateFloydTriangleWithOneRow() {
List<List<Integer>> expectedOutput = singletonList(singletonList(1));
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(1));
 }

@Test
publicvoidtestGenerateFloydTriangleWithZeroRows() {
List<List<Integer>> expectedOutput = emptyList();
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(0));
 }

@Test
publicvoidtestGenerateFloydTriangleWithNegativeRows() {
List<List<Integer>> expectedOutput = emptyList();
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(-3));
 }

@Test
publicvoidtestGenerateFloydTriangleWithMultipleRows() {
List<List<Integer>> expectedOutput = Arrays.asList(singletonList(1), Arrays.asList(2, 3), Arrays.asList(4, 5, 6), Arrays.asList(7, 8, 9, 10), Arrays.asList(11, 12, 13, 14, 15));
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(5));
 }

@Test
publicvoidtestGenerateFloydTriangleWithMoreMultipleRows() {
List<List<Integer>> expectedOutput = Arrays.asList(singletonList(1), Arrays.asList(2, 3), Arrays.asList(4, 5, 6), Arrays.asList(7, 8, 9, 10), Arrays.asList(11, 12, 13, 14, 15), Arrays.asList(16, 17, 18, 19, 20, 21), Arrays.asList(22, 23, 24, 25, 26, 27, 28));
assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(7));
 }
}