aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/store-query-viatra/src/test/java/tools/refinery/store/query/viatra/internal/cardinality/UpperCardinalitySumAggregationOperatorStreamTest.java
blob: c529117e0834c55614459c7d0d50c1e867adf837 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package tools.refinery.store.query.viatra.internal.cardinality;

import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import tools.refinery.store.representation.cardinality.UpperCardinalities;
import tools.refinery.store.representation.cardinality.UpperCardinality;

import java.util.List;
import java.util.stream.Stream;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;

class UpperCardinalitySumAggregationOperatorStreamTest {
	@ParameterizedTest
	@MethodSource
	void testStream(List<UpperCardinality> list, UpperCardinality expected) {
		var result = UpperCardinalitySumAggregationOperator.INSTANCE.aggregateStream(list.stream());
		assertThat(result, equalTo(expected));
	}

	static Stream<Arguments> testStream() {
		return Stream.of(
				Arguments.of(List.of(), UpperCardinalities.ZERO),
				Arguments.of(List.of(UpperCardinality.of(3)), UpperCardinality.of(3)),
				Arguments.of(
						List.of(
								UpperCardinality.of(2),
								UpperCardinality.of(3)
						),
						UpperCardinality.of(5)
				),
				Arguments.of(List.of(UpperCardinalities.UNBOUNDED), UpperCardinalities.UNBOUNDED),
				Arguments.of(
						List.of(
								UpperCardinalities.UNBOUNDED,
								UpperCardinalities.UNBOUNDED
						),
						UpperCardinalities.UNBOUNDED
				),
				Arguments.of(
						List.of(
								UpperCardinalities.UNBOUNDED,
								UpperCardinality.of(3)
						),
						UpperCardinalities.UNBOUNDED
				)
		);
	}
}