@Test
public void should_create_next_bert_block_option() {

    assertEquals(new BlockOption(2, S_512, true),
            BlockWiseCallback.nextBertBlock(new BlockOption(1, S_512, true), 2000, 1, 1000));
    assertEquals(new BlockOption(4, S_512, false),
            BlockWiseCallback.nextBertBlock(new BlockOption(3, S_512, true), 2000, 1, 1000));
    //BERT
    assertEquals(new BlockOption(2, S_1024_BERT, true),
            BlockWiseCallback.nextBertBlock(new BlockOption(0, S_1024_BERT, true), 10000, 2, 3000));
}