InsertIntervalTest.java

import org.example.problems.insert_interval.Solution;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.Arrays;
import java.util.stream.Stream;

public class InsertIntervalTest {
    private static Stream<Arguments> provideCases() {
        return Stream.of(
            Arguments.of(new int[][]{{1, 3}, {6, 9}}, new int[]{2, 5}, new int[][]{{1, 5}, {6, 9}}),
            Arguments.of(
                new int[][]{{1, 2}, {3, 5}, {6, 7}, {8, 10}, {12, 16}},
                new int[]{4, 8},
                new int[][]{{1, 2}, {3, 10}, {12, 16}}
            )
        );
    }

    @ParameterizedTest
    @MethodSource("provideCases")
    public void test(int[][] intervals, int[] newInterval, int[][] expected) {
        Solution s = new Solution();
        Assertions.assertTrue(Arrays.deepEquals(expected, s.insert(intervals, newInterval)));
    }
}