def test_upgrade_in_black_smith(self):
player1 = Player('Humans', 'p1')
blacksmith = Blacksmith()
blacksmith.owner = player1
blacksmith.upgrade_in_blacksmith('melee_attack_l2')
self.assertEqual(player1.unit_upgrade_group['melee_attack_l2'], False)
player1.create_unit(Keep)
blacksmith.upgrade_in_blacksmith('melee_attack_l2')
self.assertEqual(player1.unit_upgrade_group['melee_attack_l2'],True)
def test_create_unit(self):
player1 = Player('Humans', 'p1')
self.assertEqual(len(player1.unit_list),0)
self.assertEqual(len(player1.tech_tree_list),0)
player1.create_unit(Peasant)
list = []
for unit in player1.unit_list:
list.append(unit[:7])
self.assertEqual(list,['peasant'])
self.assertEqual(player1.tech_tree_list,['peasant'])
print(player1.unit_instance)
def test_creat_unit_in_barracks(self):
player1 = Player('Humans', 'p1')
barracks = Barracks()
barracks.owner = player1
self.assertEqual(player1.tech_tree_list, [])
barracks.creat_unit_in_barracks(Footman)
player1.create_unit(Peasant)
self.assertEqual(player1.tech_tree_list,['footman','peasant'])
list = []
for unit in player1.unit_list:
list.append(unit[:7])
self.assertEqual(list, ['footman','peasant'])
barracks.creat_unit_in_barracks(Knight)
list = []
for unit in player1.unit_list:
list.append(unit[:7])
self.assertEqual(list, ['footman','peasant'])
player1.create_unit(Castle)
barracks.creat_unit_in_barracks(Knight)
list = []
for unit in player1.unit_list:
list.append(unit[:6])
self.assertEqual(list, ['footma','peasan','castle','knight'])
