use of aima.core.environment.wumpusworld.AgentPosition in project aima-java by aimacode.
the class HybridWumpusAgentTest method testPlanShot.
@SuppressWarnings("serial")
@Test
public void testPlanShot() {
HybridWumpusAgent hwa = new HybridWumpusAgent(4);
// Should be just shoot as are facing the Wumpus
Assert.assertEquals(Arrays.asList(new Shoot()), hwa.planShot(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(3, 1));
}
}, allRooms(4)));
Assert.assertEquals(Arrays.asList(new TurnLeft(AgentPosition.Orientation.FACING_EAST), new Shoot()), hwa.planShot(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(1, 2));
}
}, allRooms(4)));
Assert.assertEquals(Arrays.asList(new Forward(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST)), new TurnLeft(AgentPosition.Orientation.FACING_EAST), new Shoot()), hwa.planShot(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(2, 2));
}
}, allRooms(4)));
}
use of aima.core.environment.wumpusworld.AgentPosition in project aima-java by aimacode.
the class WumpusKnowledgeBaseTest method testAskPossibleWumpusRooms.
@SuppressWarnings("serial")
@Test
public void testAskPossibleWumpusRooms() {
WumpusKnowledgeBase KB;
int t = 0;
KB = new WumpusKnowledgeBase(dpll, 2);
step(KB, new AgentPercept(false, false, false, false, false), t);
Assert.assertEquals(new HashSet<Room>() {
{
add(new Room(2, 2));
}
}, KB.askPossibleWumpusRooms(t));
KB = new WumpusKnowledgeBase(dpll, 2);
step(KB, new AgentPercept(true, false, false, false, false), t);
Assert.assertEquals(new HashSet<Room>() {
{
add(new Room(1, 2));
add(new Room(2, 1));
}
}, KB.askPossibleWumpusRooms(t));
KB = new WumpusKnowledgeBase(dpll, 3);
step(KB, new AgentPercept(false, false, false, false, false), t);
Assert.assertEquals(new HashSet<Room>() {
{
add(new Room(1, 3));
add(new Room(2, 2));
add(new Room(2, 3));
add(new Room(3, 1));
add(new Room(3, 2));
add(new Room(3, 3));
}
}, KB.askPossibleWumpusRooms(t));
// Move agent to [2,1]
KB.makeActionSentence(new Forward(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST)), t);
}
use of aima.core.environment.wumpusworld.AgentPosition in project aima-java by aimacode.
the class WumpusKnowledgeBaseTest method testAskUnvistedRooms.
@SuppressWarnings("serial")
@Test
public void testAskUnvistedRooms() {
WumpusKnowledgeBase KB;
int t = 0;
KB = new WumpusKnowledgeBase(dpll, 2);
step(KB, new AgentPercept(false, false, false, false, false), t);
Assert.assertEquals(new HashSet<Room>() {
{
add(new Room(1, 2));
add(new Room(2, 1));
add(new Room(2, 2));
}
}, KB.askUnvisitedRooms(t));
// Move agent to [2,1]
KB.makeActionSentence(new Forward(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST)), t);
t++;
// NOTE: Wumpus in [2,2] so have stench
step(KB, new AgentPercept(true, false, false, false, false), t);
Assert.assertEquals(new HashSet<Room>() {
{
add(new Room(1, 2));
add(new Room(2, 2));
}
}, KB.askUnvisitedRooms(t));
}
use of aima.core.environment.wumpusworld.AgentPosition in project aima-java by aimacode.
the class WumpusKnowledgeBaseTest method testExampleInSection7_2_described_pg268_AIMA3e.
@Test
public void testExampleInSection7_2_described_pg268_AIMA3e() {
// Make smaller in order to reduce the inference time required, this still covers all the relevant rooms for the example
WumpusKnowledgeBase KB = new WumpusKnowledgeBase(dpll, 3);
int t = 0;
// 0
step(KB, new AgentPercept(false, false, false, false, false), t);
KB.makeActionSentence(new Forward(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST)), t);
// 1
t++;
step(KB, new AgentPercept(false, true, false, false, false), t);
KB.makeActionSentence(new TurnRight(AgentPosition.Orientation.FACING_EAST), t);
// 2
t++;
step(KB, new AgentPercept(false, true, false, false, false), t);
KB.makeActionSentence(new TurnRight(AgentPosition.Orientation.FACING_SOUTH), t);
// 3
t++;
step(KB, new AgentPercept(false, true, false, false, false), t);
KB.makeActionSentence(new Forward(new AgentPosition(2, 1, AgentPosition.Orientation.FACING_WEST)), t);
// 4
t++;
step(KB, new AgentPercept(false, false, false, false, false), t);
KB.makeActionSentence(new TurnRight(AgentPosition.Orientation.FACING_WEST), t);
// 5
t++;
step(KB, new AgentPercept(false, false, false, false, false), t);
KB.makeActionSentence(new Forward(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_NORTH)), t);
// 6
t++;
step(KB, new AgentPercept(true, false, false, false, false), t);
Assert.assertTrue(KB.ask(KB.newSymbol(WumpusKnowledgeBase.LOCATION, t, 1, 2)));
Assert.assertTrue(KB.ask(KB.newSymbol(WumpusKnowledgeBase.WUMPUS, 1, 3)));
Assert.assertTrue(KB.ask(KB.newSymbol(WumpusKnowledgeBase.PIT, 3, 1)));
Assert.assertTrue(KB.ask(KB.newSymbol(WumpusKnowledgeBase.OK_TO_MOVE_INTO, t, 2, 2)));
}
use of aima.core.environment.wumpusworld.AgentPosition in project aima-java by aimacode.
the class HybridWumpusAgentTest method testPlanRoute.
@SuppressWarnings("serial")
@Test
public void testPlanRoute() {
HybridWumpusAgent hwa = new HybridWumpusAgent(4);
// Should be a NoOp plan as we are already at the goal.
Assert.assertEquals(Collections.<Action>emptyList(), hwa.planRoute(new AgentPosition(1, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(1, 1));
}
}, allRooms(4)));
Assert.assertEquals(Arrays.asList(new TurnLeft(AgentPosition.Orientation.FACING_EAST), new TurnLeft(AgentPosition.Orientation.FACING_NORTH), new Forward(new AgentPosition(2, 1, AgentPosition.Orientation.FACING_WEST))), hwa.planRoute(new AgentPosition(2, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(1, 1));
}
}, allRooms(4)));
Assert.assertEquals(Arrays.asList(new TurnLeft(AgentPosition.Orientation.FACING_EAST), new Forward(new AgentPosition(3, 1, AgentPosition.Orientation.FACING_NORTH)), new Forward(new AgentPosition(3, 2, AgentPosition.Orientation.FACING_NORTH)), new TurnLeft(AgentPosition.Orientation.FACING_NORTH), new Forward(new AgentPosition(3, 3, AgentPosition.Orientation.FACING_WEST)), new Forward(new AgentPosition(2, 3, AgentPosition.Orientation.FACING_WEST)), new TurnLeft(AgentPosition.Orientation.FACING_WEST), new Forward(new AgentPosition(1, 3, AgentPosition.Orientation.FACING_SOUTH)), new Forward(new AgentPosition(1, 2, AgentPosition.Orientation.FACING_SOUTH))), hwa.planRoute(new AgentPosition(3, 1, AgentPosition.Orientation.FACING_EAST), new LinkedHashSet<Room>() {
{
add(new Room(1, 1));
}
}, new LinkedHashSet<Room>() {
{
addAll(allRooms(4));
remove(new Room(2, 1));
remove(new Room(2, 2));
}
}));
}
Aggregations