### Question:
FeaturesLabels { public long getBatchSize() { return features.shape()[0]; } FeaturesLabels(INDArray features); long getBatchSize(); void putLabels(String name, INDArray labels); INDArray getLabels(String name);  }### Answer:
@Test public void when_getBatchSizeIsCalled_expect_batchSizeIsReturned() { INDArray features = Nd4j.create(5, 10); FeaturesLabels sut = new FeaturesLabels(features); long batchSize = sut.getBatchSize(); assertEquals(5, batchSize); }