def game_loop(t): #(turns)
#this loop puts one unit against another until destroyed or out of turns.
#need to change it so unit take turns to attack each other
num_turns = t
while num_turns > 0: # while win == false:
player = Unit("") # name
enemy = Unit("") # enemy
print("Number of turns:", num_turns)
player.name=input("Unit 1 name: ")
enemy.name=input("Unit 2 name: ")
while enemy.is_alive(): # ==True
player.unit_status()
enemy.unit_status()
dice = dice_roll() #10 side die
result = player.attack(dice, enemy)
num_turns -=1
f= open("results.txt", "a+")
f.write(f"attack result {result} \r\n")
f.close()
else:
print(enemy.name, "Destroyed!")
else:
print("Game Over")
class unit_test(unittest.TestCase):
def setUp(self):
self.unit = Unit(health=100, mana=100, mana_regeneration_rate=5)
def test_init(self):
self.assertEqual(self.unit.health, 100)
self.assertEqual(self.unit.mana, 100)
self.assertEqual(self.unit._max_health, 100)
self.assertEqual(self.unit._max_mana, 100)
self.assertEqual(self.unit.mana_regeneration, 5)
def test_get_health(self):
self.assertEqual(self.unit.get_health(), 100)
def test_get_mana(self):
self.assertEqual(self.unit.get_mana(), 100)
def test_is_alive(self):
self.unit.health = 0
self.assertFalse(self.unit.is_alive())
def test_take_healing(self):
self.unit.health = 0
self.assertFalse(self.unit.take_healing(30))
self.unit.health = 50
self.unit.take_healing(100)
self.assertEqual(self.unit.health, 100)
self.unit.take_healing(-50)
self.assertEqual(self.unit.get_health(), 100)
def test_take_mana(self):
self.unit.mana = 0
self.unit.take_mana(moved=True)
self.assertEqual(self.unit.mana, 5)
self.unit.take_mana(120)
self.assertEqual(self.unit.mana, 100)
def test_equip_weapon(self):
weapon = Weapon("Weapon name", 20)
self.unit.equip(weapon)
self.assertEqual(self.unit.weapon, weapon)
def test_learn_spell(self):
spell = Spell("Spell name", 100, 50, 3)
self.unit.learn(spell)
self.assertEqual(self.unit.spell, spell)
def test_attack_without_weapon_or_spell(self):
self.assertEqual(self.unit.attack()[1], 0)
def test_attack_with_better_weapon(self):
spell = Spell("S", 10, 20, 5)
weapon = Weapon('W', 50)
self.unit.learn(spell)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack()[1], 50)
def test_attack_with_equal_weapon_and_spell(self):
spell = Spell("S", 50, 50, 5)
weapon = Weapon('W', 50)
self.unit.learn(spell)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack()[1], 50)
self.assertEqual(self.unit.mana, 50)
# def test_cant_attack_with_weapon_from_distance(self):
# weapon = Weapon('W', 50)
# self.unit.equip(weapon)
# # self.assertEqual(self.unit.attack(distance=2), 0)
# self.assertEqual(self.unit.attack()[1], 0)
def test_attack_by_weapon(self):
weapon = Weapon("Weapon name", 20)
self.assertEqual(self.unit.attack(by="weapon")[1], 0)
self.unit.equip(weapon)
self.assertEqual(self.unit.attack(by="weapon")[1], 20)
def test_attack_by_spell(self):
spell = Spell("Spell name", 100, 50, 3)
self.assertEqual(self.unit.attack(by="spell")[1], 0)
self.unit.learn(spell)
# self.assertEqual(self.unit.attack(by="spell"), 100)
def test_take_damage(self):
self.unit.take_damage(20)
self.assertEqual(self.unit.health, 80)
self.unit.take_damage(90)
self.assertEqual(self.unit.health, 0)
def test_can_cast(self):
# self.assertFalse(self.unit.can_cast())
self.unit.learn(Spell('a', 10, 10, 3))
# self.assertTrue(self.unit.can_cast(1))
# self.assertTrue(self.unit.can_cast(3))
# self.assertFalse(self.unit.can_cast(4))
self.assertTrue(self.unit.can_cast())
self.unit.mana = 5
# self.assertFalse(self.unit.can_cast(1))
self.assertFalse(self.unit.can_cast())
