TwoSumTest.java

import org.example.problems.two_sum.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.stream.Stream;

public class TwoSumTest {
    private static Stream<Arguments> provideCasesForTwoSum() {
        return Stream.of(
            Arguments.of(new int[]{2, 7, 11, 15}, 9, new int[]{0, 1}),
            Arguments.of(new int[]{3, 2, 4}, 6, new int[]{1, 2}),
            Arguments.of(new int[]{3, 3}, 6, new int[]{0, 1})
        );
    }


    @ParameterizedTest
    @MethodSource("provideCasesForTwoSum")
    public void testTwoSum(int[] nums, int target, int[] expected) {
        Solution solution = new Solution();
        Assertions.assertArrayEquals(expected, solution.twoSum(nums, target));
    }
}