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'])