### Question:
NonRecurrentActorCriticHelper extends ActorCriticHelper { @Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(trainingBatchSize, 1); } NonRecurrentActorCriticHelper(int actionSpaceSize); @Override INDArray createValueLabels(int trainingBatchSize); @Override INDArray createPolicyLabels(int trainingBatchSize); @Override void setPolicy(INDArray policy, long idx, int action, double advantage);  }### Answer:
@Test public void when_callingCreateValueLabels_expect_INDArrayWithCorrectShape() { INDArray result = sut.createValueLabels(4); assertArrayEquals(new long[] { 4, 1 }, result.shape()); }