@Test public void testDisplayData() { Top.Natural<Integer> comparer = new Top.Natural<>(); Combine.Globally<Integer, Integer> max = Max.globally(comparer); assertThat(DisplayData.from(max), hasDisplayItem("comparer", comparer.getClass())); } }
public static void addCountingAsserts(PCollection<Long> input, long start, long end) { // Count == numElements PAssert.thatSingleton(input.apply("Count", Count.globally())).isEqualTo(end - start); // Unique count == numElements PAssert.thatSingleton(input.apply(Distinct.create()).apply("UniqueCount", Count.globally())) .isEqualTo(end - start); // Min == start PAssert.thatSingleton(input.apply("Min", Min.globally())).isEqualTo(start); // Max == end-1 PAssert.thatSingleton(input.apply("Max", Max.globally())).isEqualTo(end - 1); }
public static void addCountingAsserts(PCollection<Long> input, long numElements) { // Count == numElements PAssert.thatSingleton(input.apply("Count", Count.globally())).isEqualTo(numElements); // Unique count == numElements PAssert.thatSingleton(input.apply(Distinct.create()).apply("UniqueCount", Count.globally())) .isEqualTo(numElements); // Min == 0 PAssert.thatSingleton(input.apply("Min", Min.globally())).isEqualTo(0L); // Max == numElements-1 PAssert.thatSingleton(input.apply("Max", Max.globally())).isEqualTo(numElements - 1); }
public static void addCountingAsserts( PCollection<Long> input, long count, long uniqueCount, long min, long max) { PAssert.thatSingleton(input.apply("Count", Count.globally())).isEqualTo(count); PAssert.thatSingleton(input.apply(Distinct.create()).apply("UniqueCount", Count.globally())) .isEqualTo(uniqueCount); PAssert.thatSingleton(input.apply("Min", Min.globally())).isEqualTo(min); PAssert.thatSingleton(input.apply("Max", Max.globally())).isEqualTo(max); }
@Test @Category(NeedsRunner.class) public void testBoundedToUnboundedSourceAdapter() throws Exception { long numElements = 100; BoundedSource<Long> boundedSource = CountingSource.upTo(numElements); UnboundedSource<Long, Checkpoint<Long>> unboundedSource = new BoundedToUnboundedSourceAdapter<>(boundedSource); PCollection<Long> output = p.apply(Read.from(unboundedSource).withMaxNumRecords(numElements)); // Count == numElements PAssert.thatSingleton(output.apply("Count", Count.globally())).isEqualTo(numElements); // Unique count == numElements PAssert.thatSingleton(output.apply(Distinct.create()).apply("UniqueCount", Count.globally())) .isEqualTo(numElements); // Min == 0 PAssert.thatSingleton(output.apply("Min", Min.globally())).isEqualTo(0L); // Max == numElements-1 PAssert.thatSingleton(output.apply("Max", Max.globally())).isEqualTo(numElements - 1); p.run(); }