|
| 1 | +package com.thealgorithms.maths; |
| 2 | + |
| 3 | +import static org.junit.jupiter.api.Assertions.assertArrayEquals; |
| 4 | + |
| 5 | +import java.util.stream.Stream; |
| 6 | +import org.junit.jupiter.params.ParameterizedTest; |
| 7 | +import org.junit.jupiter.params.provider.MethodSource; |
| 8 | + |
| 9 | +public class ConvolutionTest { |
| 10 | + |
| 11 | + record ConvolutionTestCase(String description, double[] signalA, double[] signalB, double[] expected) { |
| 12 | + } |
| 13 | + |
| 14 | + @ParameterizedTest(name = "{0}") |
| 15 | + @MethodSource("provideTestCases") |
| 16 | + void testConvolution(ConvolutionTestCase testCase) { |
| 17 | + double[] result = Convolution.convolution(testCase.signalA, testCase.signalB); |
| 18 | + assertArrayEquals(testCase.expected, result, 1e-9, testCase.description); |
| 19 | + } |
| 20 | + |
| 21 | + private static Stream<ConvolutionTestCase> provideTestCases() { |
| 22 | + return Stream.of(new ConvolutionTestCase("Basic convolution", new double[] {1, 2, 3}, new double[] {4, 5, 6}, new double[] {4, 13, 28, 27, 18}), new ConvolutionTestCase("Convolution with zero elements", new double[] {0, 0, 0}, new double[] {1, 2, 3}, new double[] {0, 0, 0, 0, 0}), |
| 23 | + new ConvolutionTestCase("Convolution with single element", new double[] {2}, new double[] {3}, new double[] {6}), new ConvolutionTestCase("Convolution with different sizes", new double[] {1, 2}, new double[] {3, 4, 5}, new double[] {3, 10, 13, 10}), |
| 24 | + new ConvolutionTestCase("Convolution with negative values", new double[] {1, -2, 3}, new double[] {-1, 2, -3}, new double[] {-1, 4, -10, 12, -9}), |
| 25 | + new ConvolutionTestCase("Convolution with large numbers", new double[] {1e6, 2e6}, new double[] {3e6, 4e6}, new double[] {3e12, 1e13, 8e12})); |
| 26 | + } |
| 27 | +} |
0 commit comments