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