### Question:
NonRecurrentNStepQLearningHelper extends NStepQLearningHelper { @Override public INDArray createLabels(int trainingBatchSize) { return Nd4j.create(trainingBatchSize, actionSpaceSize); } NonRecurrentNStepQLearningHelper(int actionSpaceSize); @Override INDArray createLabels(int trainingBatchSize); @Override INDArray getExpectedQValues(INDArray allExpectedQValues, int idx); @Override void setLabels(INDArray labels, long idx, INDArray data); @Override INDArray getTargetExpectedQValuesOfLast(IOutputNeuralNet target, List<StateActionPair<Integer>> trainingBatch, INDArray features);  }### Answer:
@Test public void when_callingCreateValueLabels_expect_INDArrayWithCorrectShape() { INDArray result = sut.createLabels(4); assertArrayEquals(new long[] { 4, 3 }, result.shape()); }