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));
 }
}