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)]))
