def test_team_remove_unlisted():
# Test that if no results found return 0
team = Team("One")
jodie = Hero("Jodie Foster")
team.add_hero(jodie)
code = team.remove_hero("Athena")
assert code == 0
def test_team_remove_hero():
team = Team("One")
jodie = Hero("Jodie Foster")
team.add_hero(jodie)
assert team.heroes[0].name == "Jodie Foster"
team.remove_hero("Jodie Foster")
assert len(team.heroes) == 0
def test_print_heroes():
team = Team("One")
jodie = Hero("Jodie Foster")
team.add_hero(jodie)
athena = Hero("Athena")
team.add_hero(athena)
output_string = capture_console_output(team.view_all_heroes)
assert "Jodie Foster" in output_string
assert "Athena" in output_string
class Arena:
def __init__(self):
'''Instantiate properties
team_one: None
team_two: None
'''
self.team_one = Team("Team One")
self.team_two = Team("Team Two")
def create_ability(self):
name = input("What is the ability name?")
max_damage = input("What is the max damage of the ability?")
return Ability(name, max_damage)
def create_weapon(self):
name_of_weapon = input("What is this weapons name?")
damage_of_weapon = input("What is the weapons max damage ?")
return Weapon(name_of_weapon, damage_of_weapon)
def create_armor(self):
name_of_armor = input("What is this armors name ?")
max_health = input("What is the armors max health?")
return Weapon(name_of_armor, max_health)
def create_hero(self):
hero_name = input("Hero's name: ")
hero = Hero(hero_name)
add_item = None
while add_item != "4":
add_item = input(
"[1] Add ability\n[2] Add weapon\n[3] Add armor\n[4] Done adding items\n\nYour choice: "
)
if add_item == "1":
ability = self.create_ability()
hero.add_ability(ability)
elif add_item == "2":
weapon = self.create_weapon()
hero.add_weapon(weapon)
elif add_item == "3":
armor = self.create_armor()
hero.add_armor(armor)
return hero
# build_team_one is provided to you
def build_team_one(self):
'''Prompt the user to build team_one '''
numOfTeamMembers = int(
input("How many members would you like on Team One?\n"))
for hero in range(numOfTeamMembers):
hero = self.create_hero()
self.team_one.add_hero(hero)
def build_team_two(self):
'''Prompt the user to build team_two'''
numOfTeamMembers = int(
input("How many members would you like on Team Two?\n"))
for hero in range(numOfTeamMembers):
hero = self.create_hero()
self.team_two.add_hero(hero)
def team_battle(self):
self.team_one.attack(self.team_two)
def show_stats(self):
print("\n")
print(self.team_one.name + " statistics: ")
self.team_one.stats()
print("\n")
print(self.team_two.name + " statistics: ")
self.team_two.stats()
print("\n")
# This is how to calculate the average K/D for Team One
team_kills = 0
team_deaths = 0
for hero in self.team_one.heroes:
team_kills += hero.kills
team_deaths += hero.deaths
if team_deaths == 0:
team_deaths = 1
print(self.team_one.name + " average K/D was: " +
str(team_kills / team_deaths))
team_kills = 0
team_deaths = 0
for hero in self.team_two.heroes:
team_kills += hero.kills
team_deaths += hero.deaths
if team_deaths == 0:
team_deaths = 1
print(self.team_two.name + " average K/D was: " +
str(team_kills / team_deaths))
# Here is a way to list the heroes from Team One that survived
for hero in self.team_one.heroes:
if hero.deaths == 0:
print("survived from " + self.team_one.name + ": " + hero.name)
for hero in self.team_two.heroes:
if hero.deaths == 0:
print("survived from " + self.team_two.name + ": " + hero.name)
