First clear solution for The Vampires by johankor - python coding challenges

Enable Javascript in your browser and then refresh this page, for a much enhanced experience.

First solution in Clear category for The Vampires by johankor

class Warrior:
    def __init__(self):
        self.health=50
        self.attack=5
        self.defense = 0
        self.is_alive = True


class Knight(Warrior):
    def __init__(self):
        self.health = 50
        self.attack=7
        self.defense = 0
        self.is_alive = True

    
class Defender(Warrior):
    def __init__(self):
        self.health = 60
        self.attack = 3
        self.defense = 2
        self.is_alive = True
        
        
class Rookie(Warrior):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.health = 50
        self.attack = 1
        self.defense = 0
        self.is_alive = True

class Vampire(Warrior):
    def __init__(self):
        self.health = 40
        self.attack = 4
        self.defense = 0
        self.vampirism = 0.5
        self.is_alive = True
                
        
class Army:
    def __init__(self):
        self.units=[]
    
    def add_units(self,unit,amount):
        for k in range(amount):
            if unit== Warrior:
                self.units.append(Warrior())
            elif unit == Knight:
                self.units.append(Knight())
            elif unit == Defender:
                self.units.append(Defender())
            else:
                self.units.append(Rookie())
                    
    def get_unit(self):
        unit = self.units[0]
        self.units.remove(self.units[0])
        return unit
    
    def size(self):
        return len(self.units)
    

class Battle:
        
    def fight(self, army1, army2):
        first=army1.get_unit()
        second=army2.get_unit()
        while True:
            fight_result=fight(first, second)
            if fight_result ==True and army2.size() > 0:
                second=army2.get_unit()
            elif fight_result == True and army2.size() == 0:
                return True
            elif fight_result == False and army1.size() > 0:
                first=army1.get_unit()
            elif fight_result == False and army1.size() == 0:
                return False

def fight(unit1,unit2):
    while True:
        if unit1.attack > unit2.defense:
            unit2.health=unit2.health-(unit1.attack-unit2.defense)
            if unit1 == Vampire:
                unit1.health += unit1.health + (unit1.attack -unit2.defense)*unit1.vampirism
        if unit2.health <= 0:
            unit2.is_alive = False
            return True
        if unit2.attack > unit1.defense:
            unit1.health=unit1.health-(unit2.attack-unit1.defense)
            if unit2 == Vampire:
                unit2.health += unit2.health + (unit2.attack -unit1.defense)*unit2.vampirism
        if unit1.health <= 0:
            unit1.is_alive = False
            return False


if __name__ == '__main__':
    #These "asserts" using only for self-checking and not necessary for auto-testing
    
    #fight tests
    chuck = Warrior()
    bruce = Warrior()
    carl = Knight()
    dave = Warrior()
    mark = Warrior()
    bob = Defender()
    mike = Knight()
    rog = Warrior()
    lancelot = Defender()

    assert fight(chuck, bruce) == True
    assert fight(dave, carl) == False
    assert chuck.is_alive == True
    assert bruce.is_alive == False
    assert carl.is_alive == True
    assert dave.is_alive == False
    assert fight(carl, mark) == False
    assert carl.is_alive == False
    assert fight(bob, mike) == False
    assert fight(lancelot, rog) == True

    #battle tests
    my_army = Army()
    my_army.add_units(Defender, 1)
    
    enemy_army = Army()
    enemy_army.add_units(Warrior, 2)

    army_3 = Army()
    army_3.add_units(Warrior, 1)
    army_3.add_units(Defender, 1)

    army_4 = Army()
    army_4.add_units(Warrior, 2)

    battle = Battle()

    assert battle.fight(my_army, enemy_army) == False
    assert battle.fight(army_3, army_4) == True
    print("Coding complete? Let's try tests!")


if __name__ == '__main__':
    #These "asserts" using only for self-checking and not necessary for auto-testing
    
    #fight tests
    chuck = Warrior()
    bruce = Warrior()
    carl = Knight()
    dave = Warrior()
    mark = Warrior()
    bob = Defender()
    mike = Knight()
    rog = Warrior()
    lancelot = Defender()
    eric = Vampire()
    adam = Vampire()
    richard = Defender()
    ogre = Warrior()

    assert fight(chuck, bruce) == True
    assert fight(dave, carl) == False
    assert chuck.is_alive == True
    assert bruce.is_alive == False
    assert carl.is_alive == True
    assert dave.is_alive == False
    assert fight(carl, mark) == False
    assert carl.is_alive == False
    assert fight(bob, mike) == False
    assert fight(lancelot, rog) == True
    assert fight(eric, richard) == False
    assert fight(ogre, adam) == True

    #battle tests
    my_army = Army()
    my_army.add_units(Defender, 2)
    my_army.add_units(Vampire, 2)
    my_army.add_units(Warrior, 1)
    
    enemy_army = Army()
    enemy_army.add_units(Warrior, 2)
    enemy_army.add_units(Defender, 2)
    enemy_army.add_units(Vampire, 3)

    army_3 = Army()
    army_3.add_units(Warrior, 1)
    army_3.add_units(Defender, 4)

    army_4 = Army()
    army_4.add_units(Vampire, 3)
    army_4.add_units(Warrior, 2)

    battle = Battle()

    assert battle.fight(my_army, enemy_army) == False
    assert battle.fight(army_3, army_4) == True
    print("Coding complete? Let's try tests!")

May 25, 2019

Comments:

Oleg_Domokeev on Aug. 15, 2019, 12:40 p.m.
The code is clear. Only two remarks First: You could use standard method pop() instead new method getitem() Second: You do not need use if-statement in function addunits (for example: soldier=unit() self.units.append(soldier)

johankor on Aug. 15, 2019, 1:25 p.m.
Thank you so much for your comments. Especially your second remark is so true since the unit type is given as a parameter. I couldn't thought that at the time.

Everything is Object

0 %

You ClassRoom Demo is Ready With the new tool CheckiO ClassRoom, you will be able to analyze your students' progress, have your own leaderboard, change missions order on the map, and so on.