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