Examples with ExecutionLocation org.apache.beam.runners.dataflow.worker.graph.Nodes.ExecutionLocation used on opensource projects

Example 6 with ExecutionLocation

use of org.apache.beam.runners.dataflow.worker.graph.Nodes.ExecutionLocation in project beam by apache.

the class DeduceNodeLocationsFunctionTest method assertThatLocationIsProperlyDeduced.

private void assertThatLocationIsProperlyDeduced(Node node, ExecutionLocation expectedLocation) {
    assertThat(node, instanceOf(ParallelInstructionNode.class));
    ExecutionLocation location = ((ParallelInstructionNode) node).getExecutionLocation();
    assertEquals(location, expectedLocation);
}