def test_team_attack_deaths(): team_one = superheroes.Team("One") jodie = superheroes.Hero("Jodie Foster") aliens = superheroes.Ability("Alien Friends", 10000) jodie.add_ability(aliens) team_one.add_hero(jodie) team_two = superheroes.Team("Two") athena = superheroes.Hero("Athena") socks = superheroes.Armor("Socks", 10) athena.add_armor(socks) team_two.add_hero(athena) assert team_two.members[0].deaths == 0 team_two.kill_team() assert team_two.members[0].deaths == 1
def test_team_remove_unlisted(): # Test that if no results found return 0 team = superheroes.Team("One") jodie = superheroes.Hero("Jodie Foster") team.add_hero(jodie) code = team.remove_hero("Athena") assert code == 0
def test_print_heroes(): team = superheroes.Team("One") jodie = superheroes.Hero("Jodie Foster") team.add_hero(jodie) athena = superheroes.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
def test_team_remove_hero(): team = superheroes.Team("One") jodie = superheroes.Hero("Jodie Foster") team.add_hero(jodie) jade = superheroes.Hero("Jade Marry") team.add_hero(jade) jane = superheroes.Hero("Jane Pots") team.add_hero(jane) print("there are still {} members in the team".format(len(team.members))) assert team.members[0].name == "Jodie Foster" team.remove_hero("Jodie Foster") assert len( team.members) == 2, "there are still {} members in the team".format( len(team.members))
def test_revive_heroes(): hero = [] for _ in range(0, 20): hero.append(build_hero(4, 4, 4)) team_one = superheroes.Team("One") for hero in hero: team_one.add_hero(hero) for hero in team_one.members: hero.current_health == 12 team_one.revive_heroes() for hero in team_one.members: assert hero.current_health == 100
def create_team(hero=[]): teams = [ "Orchids", "Red", "Blue", "Cheese Steaks", "Warriors", "49ers", "Marvel", "DC", "Rat Pack", "The Little Red Riding Hoods", "Team One", "Generic Team", "X-men", "Team Two", "Golden Champions", "Vegan Protectors", "The Cardinals", "Winky Bears", "Steelsmiths", "Boilermakers", "Nincompoops" ] name = teams[random.randint(0, len(teams) - 1)] team = superheroes.Team(name) if len(hero) > 0: for member in hero: team.add_hero(member) return team
def test_team_remove_empty_list(): team = superheroes.Team("One") assert team.remove_hero("Athena") == 0
def test_team_hero(): team = superheroes.Team("One") jodie = superheroes.Hero("Jodie Foster") team.add_hero(jodie) assert len(team.members) == 1 assert team.members[0].name == "Jodie Foster"
def test_team_name(): team = superheroes.Team("One") assert team.name == "One"
def test_team_instance(): team = superheroes.Team("One") assert team