88package com .powsybl .metrix .integration .chunk ;
99
1010import com .google .common .collect .Range ;
11+ import org .assertj .core .api .Assertions ;
1112import org .junit .jupiter .api .Test ;
1213
1314import 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