def test_character_get_total_vital_status(self):
        character = Character(character_name_1, uses_magic, v_status_character)
        character.level = 7
        character.add_weapon(weapon_1)
        character.add_weapon(weapon_2)
        character.add_weapon(weapon_3)
        character.add_weapon(weapon_4)
        real_level = character.get_level_with_weapons()

        weapon_health = weapon_1.vital_status.health + weapon_2.vital_status.health + \
                        weapon_3.vital_status.health + weapon_4.vital_status.health
        weapon_attack = weapon_1.vital_status.attack + weapon_2.vital_status.attack + \
                        weapon_3.vital_status.attack + weapon_4.vital_status.attack
        weapon_defense = weapon_1.vital_status.defense + weapon_2.vital_status.defense + \
                         weapon_3.vital_status.defense + weapon_4.vital_status.defense
        weapon_magic_power = weapon_1.vital_status.magic_power + weapon_2.vital_status.magic_power + \
                         weapon_3.vital_status.magic_power + weapon_4.vital_status.magic_power
        weapon_magic_defense = weapon_1.vital_status.magic_defense + weapon_2.vital_status.magic_defense + \
                         weapon_3.vital_status.magic_defense + weapon_4.vital_status.magic_defense

        character_level_status = character.vital_status.get_vital_status_at_level(
            real_level)
        final_status = character.get_total_vital_status()

        total_health = weapon_health + character_level_status.health
        total_attack = weapon_attack + character_level_status.attack
        total_magic_power = weapon_magic_power + character_level_status.magic_power
        total_defense = weapon_defense + character_level_status.defense
        total_magic_defense = weapon_magic_defense + character_level_status.magic_defense

        self.assertEqual(final_status.health, total_health)
        self.assertEqual(final_status.attack, total_attack)
        self.assertEqual(final_status.magic_power, total_magic_power)
        self.assertEqual(final_status.defense, total_defense)
        self.assertEqual(final_status.magic_defense, total_magic_defense)
 def test_character_get_total_level(self):
     character = Character(character_name_1, uses_magic, v_status_character)
     character.add_weapon(weapon_1)
     character.add_weapon(weapon_2)
     character.add_weapon(weapon_3)
     character.add_weapon(weapon_4)
     character.level = 7
     self.assertEqual(8, character.get_level_with_weapons())