Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
stevenzwu committed Dec 6, 2023
1 parent a05d120 commit fa153d6
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@
*/
package org.apache.iceberg.flink.sink.shuffle;

import static org.assertj.core.api.Assertions.assertThat;

import java.util.Map;
import org.apache.iceberg.Schema;
import org.apache.iceberg.SortKey;
Expand All @@ -26,8 +28,6 @@
import org.junit.Before;
import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;

public class TestAggregatedStatisticsTracker {
private static final int NUM_SUBTASKS = 2;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,9 @@
*/
package org.apache.iceberg.flink.sink.shuffle;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;

import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
Expand All @@ -33,9 +36,6 @@
import org.junit.Before;
import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;

public class TestDataStatisticsCoordinator {
private static final String OPERATOR_NAME = "TestCoordinator";
private static final OperatorID TEST_OPERATOR_ID = new OperatorID(1234L, 5678L);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ public void testCheckpointAndReset() throws Exception {
checkpoint1Subtask0DataStatistic.add(keyC);
DataStatisticsEvent<MapDataStatistics, Map<SortKey, Long>>
checkpoint1Subtask0DataStatisticEvent =
DataStatisticsEvent.create(1, checkpoint1Subtask0DataStatistic, statisticsSerializer);
DataStatisticsEvent.create(1, checkpoint1Subtask0DataStatistic, statisticsSerializer);

// Handle events from operators for checkpoint 1
coordinator.handleEventFromOperator(0, 0, checkpoint1Subtask0DataStatisticEvent);
Expand All @@ -105,7 +105,7 @@ public void testCheckpointAndReset() throws Exception {
checkpoint2Subtask0DataStatistic.add(keyE);
DataStatisticsEvent<MapDataStatistics, Map<SortKey, Long>>
checkpoint2Subtask0DataStatisticEvent =
DataStatisticsEvent.create(2, checkpoint2Subtask0DataStatistic, statisticsSerializer);
DataStatisticsEvent.create(2, checkpoint2Subtask0DataStatistic, statisticsSerializer);
// Handle events from operators for checkpoint 2
coordinator.handleEventFromOperator(0, 0, checkpoint2Subtask0DataStatisticEvent);
TestDataStatisticsCoordinator.waitForCoordinatorToProcessActions(dataStatisticsCoordinator);
Expand All @@ -120,8 +120,8 @@ public void testCheckpointAndReset() throws Exception {
coordinator.resetToCheckpoint(1L, checkpoint1Bytes);
DataStatisticsCoordinator<MapDataStatistics, Map<SortKey, Long>>
restoredDataStatisticsCoordinator =
(DataStatisticsCoordinator<MapDataStatistics, Map<SortKey, Long>>)
coordinator.getInternalCoordinator();
(DataStatisticsCoordinator<MapDataStatistics, Map<SortKey, Long>>)
coordinator.getInternalCoordinator();
assertThat(dataStatisticsCoordinator).isNotEqualTo(restoredDataStatisticsCoordinator);
// Verify restored data statistics
MapDataStatistics restoredAggregateDataStatistics =
Expand Down

0 comments on commit fa153d6

Please sign in to comment.