27 lines
557 B
Java
27 lines
557 B
Java
package model.mdp;
|
|
|
|
import static org.junit.Assert.assertTrue;
|
|
import model.mdp.MDP.MODE;
|
|
|
|
import org.junit.Test;
|
|
|
|
public class ValueIterationSolverTest {
|
|
|
|
@Test
|
|
public void testSolve() {
|
|
MDPSolver solver = new ValueIterationSolver();
|
|
|
|
//solve for a score of 25 in at most 35 turns
|
|
int maxScore = 6;
|
|
int maxTurns = 10;
|
|
|
|
MDP mdp = new MDP(maxScore,maxTurns,MODE.CEIL);
|
|
Policy policy = solver.solve(mdp);
|
|
|
|
assertTrue(policy.size() >= maxScore);
|
|
assertTrue(policy.size() <= maxTurns);
|
|
|
|
System.out.println("Policy: " + policy);
|
|
}
|
|
}
|