trans/graphhopper-jsprit
1
0
Fork 0
mirror of https://github.com/graphhopper/jsprit.git synced 2020-01-24 07:45:05 +01:00
Code Issues Releases Wiki Activity

tested new methods .get/putProblemState

Browse source
This commit is contained in:
oblonski 2014-03-26 12:41:47 +01:00
parent 8c2e3ca210
commit 4e23697fc8
2 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
jsprit-core/src/main/java/jsprit/core/algorithm/state/StateManager.java
View file

@ -167,7 +167,8 @@ public class StateManager implements RouteAndActivityStateGetter, IterationStart
}
<T> T getDefaultProblemState(StateId stateId, Class<T> type){
return defaultProblemStates_.getState(stateId, type);
if(defaultProblemStates_.containsKey(stateId)) return defaultProblemStates_.getState(stateId, type);
return null;
}
/**

30
jsprit-core/src/test/java/jsprit/core/algorithm/state/StateManagerTest.java
View file

@ -143,4 +143,34 @@ public class StateManagerTest {
Capacity getCap = stateManager.getActivityState(activity, id, Capacity.class);
assertEquals(500, getCap.get(0));
}
@Test
public void whenProblemStateIsSet_itMustBeSetCorrectly(){
StateManager stateManager = new StateManager(mock(VehicleRoutingTransportCosts.class));
StateId id = StateFactory.createId("problemState");
stateManager.putProblemState(id, Boolean.class, true);
boolean problemState = stateManager.getProblemState(id, Boolean.class);
assertTrue(problemState);
}
@Test(expected=NullPointerException.class)
public void whenProblemStateIsSetAndStateManagerClearedAfterwards_itThrowsException(){
StateManager stateManager = new StateManager(mock(VehicleRoutingTransportCosts.class));
StateId id = StateFactory.createId("problemState");
stateManager.putProblemState(id, Boolean.class, true);
stateManager.clear();
@SuppressWarnings("unused")
boolean problemState = stateManager.getProblemState(id, Boolean.class);
}
@Test
public void whenProblemStateIsSetAndStateManagerClearedAfterwards_itReturnsDefault(){
StateManager stateManager = new StateManager(mock(VehicleRoutingTransportCosts.class));
StateId id = StateFactory.createId("problemState");
stateManager.putDefaultProblemState(id, Boolean.class, false);
stateManager.putProblemState(id, Boolean.class, true);
stateManager.clear();
boolean problemState = stateManager.getProblemState(id, Boolean.class);
assertFalse(problemState);
}
}