def setUp(cls): def generateViking(): cls.name = 'Harald' cls.strength = 150 cls.health = 300 return Viking(cls.name, cls.health, cls.strength) def generateSaxon(): cls.health = 60 cls.strength = 25 return Saxon(cls.health, cls.strength) cls.viking = generateViking() cls.saxon = generateSaxon() cls.war = War() cls.war.addSaxon(cls.saxon) cls.war.addViking(cls.viking)
def setUp(cls): cls.war = War()