Skip to content

Commit da6ed4c

Browse files
committed
add method getRanges to ChunkCutter
Signed-off-by: marifunf <marianne.funfrock@rte-france.com>
1 parent c45c32e commit da6ed4c

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

metrix-integration/src/main/java/com/powsybl/metrix/integration/chunk/ChunkCutter.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,10 @@ public static List<Range<Integer>> splitRange(Range<Integer> rangeToSplit, int c
4949
return rangeList;
5050
}
5151

52+
public List<Range<Integer>> getRanges() {
53+
return ranges;
54+
}
55+
5256
public int getChunkSize() {
5357
return chunkSize;
5458
}

metrix-integration/src/test/java/com/powsybl/metrix/integration/chunk/ChunkCutterTest.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
package com.powsybl.metrix.integration.chunk;
99

1010
import com.google.common.collect.Range;
11+
import org.assertj.core.api.Assertions;
1112
import org.junit.jupiter.api.Test;
1213

1314
import java.util.List;
@@ -28,14 +29,15 @@ void test() {
2829
assertEquals(Range.closed(0, 91), cutter.getChunkRange(0));
2930
assertEquals(Range.closed(92, 183), cutter.getChunkRange(1));
3031
assertEquals(Range.closed(8648, 8735), cutter.getChunkRange(94));
32+
Assertions.assertThat(cutter.getRanges()).hasSize(95);
3133
}
3234

3335
@Test
3436
void test2() {
3537
ChunkCutter cutter = new ChunkCutter(List.of(Range.closed(0, 8735)), 10000);
3638
assertEquals(1, cutter.getChunkCount());
3739
assertEquals(10000, cutter.getChunkSize());
38-
assertEquals(Range.closed(0, 8735), cutter.getChunkRange(0));
40+
assertEquals(List.of(Range.closed(0, 8735)), cutter.getRanges());
3941
}
4042

4143
@Test
@@ -46,6 +48,7 @@ void test3() {
4648
assertEquals(Range.closed(0, 0), cutter.getChunkRange(0));
4749
assertEquals(Range.closed(1, 1), cutter.getChunkRange(1));
4850
assertEquals(Range.closed(8735, 8735), cutter.getChunkRange(8735));
51+
Assertions.assertThat(cutter.getRanges()).hasSize(8736);
4952
}
5053

5154
@Test
@@ -55,5 +58,6 @@ void test4() {
5558
assertEquals(92, cutter.getChunkSize());
5659
assertEquals(Range.closed(100, 191), cutter.getChunkRange(0));
5760
assertEquals(Range.closed(192, 200), cutter.getChunkRange(1));
61+
assertEquals(List.of(Range.closed(100, 191), Range.closed(192, 200)), cutter.getRanges());
5862
}
5963
}

0 commit comments

Comments
 (0)