Examples with DisplayData org.apache.beam.sdk.transforms.display.DisplayData used on opensource projects

Example 86 with DisplayData

use of org.apache.beam.sdk.transforms.display.DisplayData in project beam by apache.

the class ProxyInvocationHandlerTest method testDisplayDataExcludesValuesAccessedButNeverSet.

@Test
public void testDisplayDataExcludesValuesAccessedButNeverSet() {
    HasDefaults options = PipelineOptionsFactory.as(HasDefaults.class);
    assertEquals("bar", options.getFoo());
    DisplayData data = DisplayData.from(options);
    assertThat(data, not(hasDisplayItem("foo")));
}

Example 87 with DisplayData

use of org.apache.beam.sdk.transforms.display.DisplayData in project beam by apache.

the class DoFnTest method testDefaultPopulateDisplayDataImplementation.

@Test
public void testDefaultPopulateDisplayDataImplementation() {
    DoFn<String, String> fn = new DoFn<String, String>() {
    };
    DisplayData displayData = DisplayData.from(fn);
    assertThat(displayData.items(), empty());
}

Example 88 with DisplayData

use of org.apache.beam.sdk.transforms.display.DisplayData in project beam by apache.

the class CombineFnsTest method testComposedCombineDisplayData.

@Test
public void testComposedCombineDisplayData() {
    SimpleFunction<String, String> extractFn = new SimpleFunction<String, String>() {

        @Override
        public String apply(String input) {
            return input;
        }
    };
    DisplayDataCombineFn combineFn1 = new DisplayDataCombineFn("value1");
    DisplayDataCombineFn combineFn2 = new DisplayDataCombineFn("value2");
    CombineFns.ComposedCombineFn<String> composedCombine = CombineFns.compose().with(extractFn, combineFn1, new TupleTag<>()).with(extractFn, combineFn2, new TupleTag<>());
    DisplayData displayData = DisplayData.from(composedCombine);
    assertThat(displayData, hasDisplayItem("combineFn1", combineFn1.getClass()));
    assertThat(displayData, hasDisplayItem("combineFn2", combineFn2.getClass()));
    assertThat(displayData, includesDisplayDataFor("combineFn1", combineFn1));
    assertThat(displayData, includesDisplayDataFor("combineFn2", combineFn2));
}

Example 89 with DisplayData

use of org.apache.beam.sdk.transforms.display.DisplayData in project beam by apache.

the class TopTest method testDisplayData.

@Test
public void testDisplayData() {
    Top.Natural<Integer> comparer = new Top.Natural<>();
    Combine.Globally<Integer, List<Integer>> top = Top.of(1234, comparer);
    DisplayData displayData = DisplayData.from(top);
    assertThat(displayData, hasDisplayItem("count", 1234));
    assertThat(displayData, hasDisplayItem("comparer", comparer.getClass()));
}

Example 90 with DisplayData

use of org.apache.beam.sdk.transforms.display.DisplayData in project beam by apache.

the class PartitionTest method testDisplayDataOfSideViewFunction.

@Test
public void testDisplayDataOfSideViewFunction() {
    Partition<?> partition = Partition.of(123, new IdentitySideViewFn(), Requirements.empty());
    DisplayData displayData = DisplayData.from(partition);
    assertThat(displayData, hasDisplayItem("numPartitions", 123));
    assertThat(displayData, hasDisplayItem("partitionFn", IdentitySideViewFn.class));
}