class ShipTests(unittest.TestCase):
def setUp(self):
self.ship = Ship((1, 1), (1, 5), 'Carrier')
def test_ship_length(self):
self.assertEqual(self.ship.length, 5)
def test_ship_occupied_coords(self):
self.assertEqual(self.ship.points[0], {'coords': (1, 1), 'mark': ' '})
self.assertEqual(self.ship.points[-1], {'coords': (1, 5), 'mark': ' '})
def test_ship_hp(self):
self.assertEqual(self.ship.hp, 5)
def test_ship_hit(self):
with self.assertRaises(ShipError):
self.ship.register_hit((2, 2))
self.ship.register_hit((1, 3))
self.assertEqual(self.ship.hp, 4)
