aljoscha commented on a change in pull request #14249:
URL: https://github.com/apache/flink/pull/14249#discussion_r531579108



##########
File path: 
flink-streaming-java/src/test/java/org/apache/flink/streaming/api/graph/StreamingJobGraphGeneratorTest.java
##########
@@ -613,6 +617,51 @@ public void testShuffleModeUndefined() {
                        
sourceAndMapVertex.getProducedDataSets().get(0).getResultType());
        }
 
+       @Test
+       public void testPartitionTypesInBatchMode() {
+               StreamExecutionEnvironment env = 
StreamExecutionEnvironment.getExecutionEnvironment();
+               env.setRuntimeMode(RuntimeExecutionMode.BATCH);
+               env.setParallelism(4);
+               env.disableOperatorChaining();
+               DataStream<Integer> source = env.fromElements(1);
+               source
+                       // set the same parallelism as the source to make it a 
FORWARD SHUFFLE
+                       .map(value -> value).setParallelism(1)
+                       .rescale()

Review comment:
       Shouldn't there also be a `rescale()` that is blocking? To check that it 
is, in fact, blocking in that case?




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to