def setUp(self):
Robot.initHardwareDevices = mock.Mock(return_value="")
Robot.moveToPosition = mock.Mock(return_value="")
Robot.setLedOn = mock.Mock(return_value="")
self.robot = Robot(Queue.Queue())
self.aGame = Game(self.robot, Queue.Queue())
self.aGame.currentState = HasNoPuckState(self.aGame)
self.aGame.listOfPucks = [
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE)
]
def setUp(self):
Robot.initHardwareDevices = mock.Mock(return_value="")
Robot.rotateToAngle = mock.Mock(return_value="")
Robot.findLetterInLetterBoard = mock.Mock(return_value="D")
Robot.setCamMiddle = mock.Mock(return_value="")
Robot.setCamUp = mock.Mock(return_value="")
NearResistanceState.fetchResistance = mock.Mock(return_value="")
self.robot = Robot(Queue.Queue())
self.aGame = Game(self.robot, Queue.Queue())
self.aGame.currentState = NearLetterBoardState(self.aGame)
self.aGame.listOfPucks = [Puck(RANDOM_COLOR_CODE), Puck(RANDOM_COLOR_CODE), Puck(RANDOM_COLOR_CODE)]
self.aGame.setPositionToReadInLetterBoard(2)
class PuckTest(unittest.TestCase):
def setUp(self):
self.aPuck = Puck(A_COLOR)
def test_whenAPuckIsCreatedPriorityShouldBeMinusOne(self):
self.assertTrue(self.aPuck.getPriority() == -1)
def setUp(self):
Robot.initHardwareDevices = mock.Mock(return_value="")
Robot.moveToPosition = mock.Mock(return_value="")
Robot.moveToRight = mock.Mock(return_value="")
Robot.moveForward = mock.Mock(return_value="")
Robot.rotateToAngle = mock.Mock(return_value="")
NearGoalPositionState.goToCorner = mock.Mock(return_value="")
Robot.findLetterInLetterBoard = mock.Mock(return_value="D")
Robot.dropPuck = mock.Mock(return_value="SA")
NearPucksState.fetchPuck = mock.Mock(return_value="")
self.robot = Robot(Queue.Queue())
self.aGame = Game(self.robot, Queue.Queue())
self.aGame.currentState = NearGoalPositionState(self.aGame)
self.aGame.listOfPucks = [
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE)
]
self.aGame.setPucksOrientation("SA")
self.aGame.setPositionToReadInLetterBoard(2)
self.aGame.listOfPucks[0].setCornerToBePlaced("D")
def setUp(self):
Robot.initHardwareDevices = mock.Mock(return_value="")
Robot.moveBackward = mock.Mock(return_value="")
Robot.grabPuck = mock.Mock(return_value="")
Robot.rotateToAngle = mock.Mock(return_value="")
Robot.rotateClockwise = mock.Mock(return_value="")
Robot.rotateCounterClockwise = mock.Mock(return_value="")
Robot.moveBackward = mock.Mock(return_value="")
Robot.moveForward = mock.Mock(return_value="")
Robot.setCamMiddle = mock.Mock(return_value="")
Robot.setCamDown = mock.Mock(return_value="")
Robot.getDistanceToPuck = mock.Mock(return_value=(20, 15))
self.robot = Robot(Queue.Queue())
self.aGame = Game(self.robot, Queue.Queue())
self.aGame.puckNumberBeingProcessed = 0
self.aGame.listOfPucks = [
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE),
Puck(RANDOM_COLOR_CODE)
]
self.aGame.currentState = NearPucksState(self.aGame)
def findCorrespondingColorsOfPucks(self, number1, number2, number3):
#Because the first and third number of the resistance are the most
#likely to be right so we want to place them first
puck1 = Puck(number1)
puck1.setPriority(1)
puck2 = Puck(number2)
puck2.setPriority(2)
puck3 = Puck(number3)
puck3.setPriority(1)
self.game.listOfPucks.append(puck1)
self.game.listOfPucks.append(puck2)
self.game.listOfPucks.append(puck3)
self.game.robotQueue.put("C1-" + COLOR_OF_PUCKS[number1])
self.game.robotQueue.put("C2-" + COLOR_OF_PUCKS[number2])
self.game.robotQueue.put("C3-" + COLOR_OF_PUCKS[number3])
def setUp(self):
self.aPuck = Puck(A_COLOR)
def test_aGameWhenNotThreePucksPlacedShouldNotBeFinished(self):
puck1 = Puck(RANDOM_COLOR_CODE)
puck1.setHasBeenPlaced(True)
puck2 = Puck(RANDOM_COLOR_CODE_1)
puck2.setHasBeenPlaced(False)
puck3 = Puck(RANDOM_COLOR_CODE_2)
puck3.setHasBeenPlaced(True)
self.game.listOfPucks.append(puck1)
self.game.listOfPucks.append(puck2)
self.game.listOfPucks.append(puck3)
self.assertFalse(self.game.isFinished())
def setUp(self):
Robot.initHardwareDevices = mock.Mock(return_value="")
Robot.moveLeft = mock.Mock(return_value="")
Robot.moveRight = mock.Mock(return_value="")
Robot.rotateToAngle = mock.Mock(return_value="")
self.robot = Robot(Queue.Queue())
self.aGame = Game(self.robot, Queue.Queue())
self.aGame.currentState = HasReadOrientationBoardState(self.aGame)
puck1 = Puck(RANDOM_COLOR_CODE)
puck1.setPriority(1)
puck2 = Puck(RANDOM_COLOR_CODE)
puck2.setPriority(2)
puck3 = Puck(RANDOM_COLOR_CODE)
puck3.setPriority(1)
self.aGame.listOfPucks = [puck1, puck2, puck3]
self.aGame.setPucksOrientation("SA")
self.aGame.listOfPucks[0].setCornerToBePlaced("D")
self.aGame.setPositionToReadInOrientationBoard(3)