### Question:
NonRecurrentNStepQLearningHelper extends NStepQLearningHelper { @Override public INDArray getExpectedQValues(INDArray allExpectedQValues, int idx) { return allExpectedQValues.getRow(idx); } 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_callingGetExpectedQValues_expect_INDArrayWithCorrectShape() { INDArray allExpectedQValues = Nd4j.create(new double[] { 1.1, 1.2, 2.1, 2.2 }).reshape(2,2); INDArray result = sut.getExpectedQValues(allExpectedQValues, 1); assertEquals(2.1, result.getDouble(0), 0.00001); assertEquals(2.2, result.getDouble(1), 0.00001); }