@Test
public void testComputeSolution() {
RandomWalk sp = new RandomWalk(new Random(0));
FSGAgentState state = new FSGAgentState(sp);
double[] solution = state.computeSolution(sp.policy(), 0.9, 0.0);
checkEquals(sp.expectedDiscountedSolution(), solution);
checkEquals(
new double[] {1 / 6.0, 2 / 6.0, 3 / 6.0, 4 / 6.0, 5 / 6.0},
state.computeSolution(sp.policy(), 1.0, 0.5));
}
