def testFleetTwoAllMarks(self):
fleet = Fleet(2)
marks = fleet.getAllImages()
self.assertEqual(len(marks), 1)
self.assertEqual(marks[0], "fleet_2_mark.png")
def testFleetOneAllMarks(self):
fleet = Fleet(1)
marks = fleet.getAllImages()
self.assertEqual(len(marks), 2)
self.assertEqual(marks[0], "fleet_1_flagship_mark.png")
self.assertEqual(marks[1], "fleet_1_mark.png")