Example #1
0
 def setUp(self):
     self.testPlayer = Player()
     self.mockPlayer = mock.MagicMock()
     self.game = Game(self.testPlayer, self.mockPlayer)
     self.game.boardWidth = BOARD_WIDTH
     self.game.boardHeight = BOARD_HEIGHT
     self.ship1 = Ship("ship1", 3, self.game)
     self.ship2 = Ship("ship2", 4, self.game)
     self.ship3 = Ship("ship3", 2, self.game)
     self.testPlayer._setShips([self.ship1, self.ship2, self.ship3])
Example #2
0
class TestPlaceShip(unittest.TestCase):
    def setUp(self):
        self.game = Game(mock.MagicMock(), mock.MagicMock())
        self.testShip = Ship('testShip1', 3, self.game)

    def test_placeShipUp(self):
        self.testShip.placeShip((0, 0), UP)
        self.assertEquals(self.testShip.locations, set([(0, 0),
                                                        (0, 1),
                                                        (0, 2),
                                                        ]))

    def test_placeShipDown(self):
        self.testShip.placeShip((5, 5), DOWN)
        self.assertEquals(self.testShip.locations, set([(5, 5),
                                                        (5, 4),
                                                        (5, 3),
                                                        ]))

    def test_placeShipRight(self):
        self.testShip.placeShip((5, 2), RIGHT)
        self.assertEquals(self.testShip.locations, set([(5, 2),
                                                        (6, 2),
                                                        (7, 2),
                                                        ]))

    def test_placeShipLeft(self):
        self.testShip.placeShip((5, 2), LEFT)
        self.assertEquals(self.testShip.locations, set([(5, 2),
                                                        (4, 2),
                                                        (3, 2),
                                                        ]))
Example #3
0
 def setUp(self):
     self.testPlayer = Player()
     self.mockPlayer = mock.MagicMock()
     self.game = Game(self.testPlayer, self.mockPlayer)
     self.game.boardWidth = BOARD_WIDTH
     self.game.boardHeight = BOARD_HEIGHT
     self.ship1 = Ship('ship1', 3, self.game)
     self.ship2 = Ship('ship2', 4, self.game)
     self.ship3 = Ship('ship3', 2, self.game)
     self.testPlayer._setShips([self.ship1,
                                self.ship2,
                                self.ship3,
                                ])
     self.ship1.placeShip((5, 5), DOWN)
     self.ship2.placeShip((0, 6), RIGHT)
     self.ship3.placeShip((6, 5), RIGHT)
Example #4
0
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.timeoutLength = 0.01
        self.game = Game(self.testPlayer, self.mockPlayer, timeoutLength=self.timeoutLength)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT

        self.testPlayer.currentGame = self.game

        self.ship1 = Ship("ship1", 3, self.game)
        self.ship2 = Ship("ship2", 4, self.game)
        self.ship3 = Ship("ship3", 2, self.game)
        self.testPlayer._setShips([self.ship1, self.ship2, self.ship3])
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)
Example #5
0
class TestIsSunk(unittest.TestCase):
    def setUp(self):
        self.game = Game(mock.MagicMock(), mock.MagicMock())
        self.testShip = Ship('testShip1', 3, self.game)
        self.testShip.placeShip((5, 5), UP)

    def test_notSunk(self):
        self.testShip.hits = set([(5, 5),
                                  (5, 6),
                                  ])
        self.assertFalse(self.testShip.isSunk())

    def test_sunk(self):
        self.testShip.hits = set([(5, 5),
                                  (5, 6),
                                  (5, 7),
                                  ])
        self.assertTrue(self.testShip.isSunk())
Example #6
0
class TestCheckAllShipsSunk(unittest.TestCase):
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.game = Game(self.testPlayer, self.mockPlayer)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT
        self.ship1 = Ship('ship1', 3, self.game)
        self.ship2 = Ship('ship2', 4, self.game)
        self.ship3 = Ship('ship3', 2, self.game)
        self.testPlayer._setShips([self.ship1,
                                   self.ship2,
                                   self.ship3,
                                   ])
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)

    def test_noShipsSunk(self):
        self.assertFalse(self.testPlayer._checkAllShipsSunk())

    def test_oneShipSunk(self):
        self.ship1.hits = self.ship1.locations
        self.assertFalse(self.testPlayer._checkAllShipsSunk())

    def test_allShipsSunk(self):
        self.ship1.hits = self.ship1.locations
        self.ship2.hits = self.ship2.locations
        self.ship3.hits = self.ship3.locations

        self.assertTrue(self.testPlayer._checkAllShipsSunk())
Example #7
0
class TestCheckIsHit(unittest.TestCase):
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.game = Game(self.testPlayer, self.mockPlayer)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT
        self.ship1 = Ship('ship1', 3, self.game)
        self.ship2 = Ship('ship2', 4, self.game)
        self.ship3 = Ship('ship3', 2, self.game)
        self.testPlayer._setShips([self.ship1,
                                   self.ship2,
                                   self.ship3,
                                   ])
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)

    def test_checkIsMiss(self):
        shot = (1, 1)
        hit, hitShip = self.testPlayer._checkIsHit(shot)

        self.assertFalse(hit)
        self.assertEqual(hitShip, None)

    def test_checkIsHit(self):
        shot = (5, 3)
        hit, hitShip = self.testPlayer._checkIsHit(shot)

        self.assertTrue(hit)
        self.assertEqual(hitShip, self.ship1)
Example #8
0
class TestGameClockViolations(unittest.TestCase):
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.timeoutLength = .01
        self.game = Game(self.testPlayer, self.mockPlayer, timeoutLength=self.timeoutLength)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT

        self.testPlayer.currentGame = self.game

        self.ship1 = Ship('ship1', 3, self.game)
        self.ship2 = Ship('ship2', 4, self.game)
        self.ship3 = Ship('ship3', 2, self.game)
        self.testPlayer._setShips([self.ship1,
                                   self.ship2,
                                   self.ship3,
                                   ])
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)

    def test_newGame(self):
        self.testPlayer.newGame = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._newGame)

    def test_placeShips(self):
        self.testPlayer.placeShips = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._placeShips)

    def test_shotHit(self):
        self.testPlayer.shotHit = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shotHit, (0, 0), 'ship1')

    def test_shotMissed(self):
        self.testPlayer.shotMissed = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shotMissed, (0, 0))

    def test_shipSunk(self):
        self.testPlayer.shipSunk = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shipSunk, 'ship1')

    def test_fireShot(self):
        self.testPlayer.fireShot = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._fireShot)

    def test_gameWon(self):
        self.testPlayer.gameWon = generateLongRunningFunc(self.timeoutLength)
        self.testPlayer._gameWon()

    def test_gameLost(self):
        self.testPlayer.gameLost = generateLongRunningFunc(self.timeoutLength)
        self.testPlayer._gameLost()

    def test_opponentShot(self):
        self.testPlayer.opponentShot = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._opponentShot, (0, 0))
Example #9
0
class TestAddHit(unittest.TestCase):
    def setUp(self):
        self.game = Game(mock.MagicMock(), mock.MagicMock())
        self.testShip = Ship('testShip1', 3, self.game)

    def test_notAHit(self):
        self.testShip.placeShip((5, 5), UP)
        shot = (0, 0)
        self.testShip.addHit(shot)

        self.assertEquals(self.testShip.hits, set())

    def test_hit(self):
        self.testShip.placeShip((5, 5), UP)
        shot = (5, 7)
        self.testShip.addHit(shot)

        self.assertEquals(self.testShip.hits, set([(5, 7)]))
Example #10
0
class TestShipsPlacedLegally(unittest.TestCase):
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.game = Game(self.testPlayer, self.mockPlayer)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT
        self.ship1 = Ship('ship1', 3, self.game)
        self.ship2 = Ship('ship2', 4, self.game)
        self.ship3 = Ship('ship3', 2, self.game)
        self.testPlayer._setShips([self.ship1,
                                   self.ship2,
                                   self.ship3,
                                   ])

    def test_shipPlacedOffBoard(self):
        self.ship1.placeShip((0, 0), DOWN)
        result = self.testPlayer.isShipPlacedLegally(self.ship1)

        self.assertFalse(result)
        self.assertFalse(self.testPlayer._allShipsPlacedLegally())

    def test_shipsOverlapping(self):
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((5, 5), RIGHT)

        result = self.testPlayer.isShipPlacedLegally(self.ship1)
        self.assertFalse(result)

        result = self.testPlayer.isShipPlacedLegally(self.ship2)
        self.assertTrue(result)

        result = self.testPlayer.isShipPlacedLegally(self.ship3)
        self.assertFalse(result)
        self.assertFalse(self.testPlayer._allShipsPlacedLegally())

    def test_shipsAreValid(self):
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)

        result = self.testPlayer.isShipPlacedLegally(self.ship1)
        self.assertTrue(result)

        result = self.testPlayer.isShipPlacedLegally(self.ship2)
        self.assertTrue(result)

        result = self.testPlayer.isShipPlacedLegally(self.ship3)
        self.assertTrue(result)
        self.assertTrue(self.testPlayer._allShipsPlacedLegally())
Example #11
0
class TestGameClockViolations(unittest.TestCase):
    def setUp(self):
        self.testPlayer = Player()
        self.mockPlayer = mock.MagicMock()
        self.timeoutLength = 0.01
        self.game = Game(self.testPlayer, self.mockPlayer, timeoutLength=self.timeoutLength)
        self.game.boardWidth = BOARD_WIDTH
        self.game.boardHeight = BOARD_HEIGHT

        self.testPlayer.currentGame = self.game

        self.ship1 = Ship("ship1", 3, self.game)
        self.ship2 = Ship("ship2", 4, self.game)
        self.ship3 = Ship("ship3", 2, self.game)
        self.testPlayer._setShips([self.ship1, self.ship2, self.ship3])
        self.ship1.placeShip((5, 5), DOWN)
        self.ship2.placeShip((0, 6), RIGHT)
        self.ship3.placeShip((6, 5), RIGHT)

    def test_newGame(self):
        self.testPlayer.newGame = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._newGame)

    def test_placeShips(self):
        self.testPlayer.placeShips = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._placeShips)

    def test_shotHit(self):
        self.testPlayer.shotHit = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shotHit, (0, 0), "ship1")

    def test_shotMissed(self):
        self.testPlayer.shotMissed = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shotMissed, (0, 0))

    def test_shipSunk(self):
        self.testPlayer.shipSunk = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._shipSunk, "ship1")

    def test_fireShot(self):
        self.testPlayer.fireShot = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._fireShot)

    def test_gameWon(self):
        self.testPlayer.gameWon = generateLongRunningFunc(self.timeoutLength)
        self.testPlayer._gameWon()

    def test_gameLost(self):
        self.testPlayer.gameLost = generateLongRunningFunc(self.timeoutLength)
        self.testPlayer._gameLost()

    def test_opponentShot(self):
        self.testPlayer.opponentShot = generateLongRunningFunc(self.timeoutLength)
        self.assertRaises(GameClockViolationException, self.testPlayer._opponentShot, (0, 0))

    def test_isValidPoint(self):
        point1 = (0, 0)
        point2 = (self.game.boardWidth - 1, 0)
        point3 = (0, self.game.boardHeight - 1)
        point4 = (self.game.boardWidth - 1, self.game.boardHeight - 1)
        point5 = (-1, 0)
        point6 = (0, -1)
        point7 = (self.game.boardWidth, self.game.boardHeight)
        point8 = (-10, -10)

        self.assertTrue(self.testPlayer._isValidPoint(point1))
        self.assertTrue(self.testPlayer._isValidPoint(point2))
        self.assertTrue(self.testPlayer._isValidPoint(point3))
        self.assertTrue(self.testPlayer._isValidPoint(point4))

        self.assertFalse(self.testPlayer._isValidPoint(point5))
        self.assertFalse(self.testPlayer._isValidPoint(point6))
        self.assertFalse(self.testPlayer._isValidPoint(point7))
        self.assertFalse(self.testPlayer._isValidPoint(point8))
Example #12
0
 def setUp(self):
     self.game = Game(mock.MagicMock(), mock.MagicMock())
     self.testShip = Ship('testShip1', 3, self.game)
     self.testShip.placeShip((5, 5), UP)
Example #13
0
 def setUp(self):
     self.game = Game(mock.MagicMock(), mock.MagicMock())
     self.testShip = Ship('testShip1', 3, self.game)
Example #14
0
class TestIsPlacementValid(unittest.TestCase):
    def setUp(self):
        self.game = Game(mock.MagicMock(), mock.MagicMock())
        self.testShip = Ship('testShip1', 3, self.game)

    def test_isPlacementValid(self):
        self.testShip.placeShip((0, 1), DOWN)
        self.assertFalse(self.testShip.isPlacementValid())

        self.testShip.placeShip((0, 2), LEFT)
        self.assertFalse(self.testShip.isPlacementValid())

        self.testShip.placeShip((10, 2), RIGHT)
        self.assertFalse(self.testShip.isPlacementValid())

        self.testShip.placeShip((5, 5), UP)
        self.assertTrue(self.testShip.isPlacementValid())

        self.testShip.placeShip((0, 0), UP)
        self.assertTrue(self.testShip.isPlacementValid())

        self.testShip.placeShip((9, 0), UP)
        self.assertTrue(self.testShip.isPlacementValid())