class TestPlaceShip(unittest.TestCase):
def setUp(self):
self.testShip = Ship('testShip1', 3)
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),
]))
class TestIsSunk(unittest.TestCase):
def setUp(self):
self.testShip = Ship('testShip1', 3)
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())
class TestAddHit(unittest.TestCase):
def setUp(self):
self.testShip = Ship('testShip1', 3)
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)]))
class TestCheckAllShipsSunk(unittest.TestCase):
def setUp(self):
self.testPlayer = Player()
self.ship1 = Ship('ship1', 3)
self.ship2 = Ship('ship2', 4)
self.ship3 = Ship('ship3', 2)
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())
class TestCheckIsHit(unittest.TestCase):
def setUp(self):
self.testPlayer = Player()
self.ship1 = Ship('ship1', 3)
self.ship2 = Ship('ship2', 4)
self.ship3 = Ship('ship3', 2)
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)
class TestShipsPlacedLegally(unittest.TestCase):
def setUp(self):
self.testPlayer = Player()
self.ship1 = Ship('ship1', 3)
self.ship2 = Ship('ship2', 4)
self.ship3 = Ship('ship3', 2)
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())
class TestIsPlacementValid(unittest.TestCase):
def setUp(self):
self.testShip = Ship('testShip1', 3)
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())
