#!/usr/bin/env python3 """ YAML单位加载器测试 测试YAML单位定义系统的各种功能 """ import pytest import sys import os sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..')) from kards_battle.units.unit_loader import UnitLoader, load_unit, list_all_units, get_unit_loader from kards_battle.units.unit import Unit from kards_battle.core.enums import UnitType class TestUnitLoader: """测试单位加载器""" def setup_method(self): """每个测试前的设置""" # 创建独立的加载器实例 self.loader = UnitLoader() def test_load_all_units_from_assets(self): """测试：从assets目录加载所有单位""" units_data = self.loader.load_all_units() # 验证加载了正确数量的单位 assert len(units_data) > 0, "应该加载至少一个单位" # 验证德军和美军单位都存在 german_units = [uid for uid in units_data.keys() if uid.startswith('ger_')] usa_units = [uid for uid in units_data.keys() if uid.startswith('usa_')] assert len(german_units) > 0, "应该有德军单位" assert len(usa_units) > 0, "应该有美军单位" def test_create_unit_from_yaml_data(self): """测试：从YAML数据创建单位实例""" self.loader.load_all_units() # 测试德军步兵 unit = self.loader.create_unit_from_id("ger_infantry_grenadier") assert isinstance(unit, Unit) assert unit.name == "German Grenadier" assert unit.unit_type == UnitType.INFANTRY assert unit.stats.attack == 2 assert unit.stats.defense == 1 assert unit.stats.operation_cost == 1 assert unit.nation == "GERMANY" assert unit.definition_id == "ger_infantry_grenadier" def test_unit_with_keywords(self): """测试：带关键词的单位加载""" self.loader.load_all_units() # 测试带BLITZ关键词的单位 unit = self.loader.create_unit_from_id("ger_infantry_15th_cavalry") assert "BLITZ" in unit.keywords assert unit.name == "15th Cavalry Regiment" def test_different_unit_types(self): """测试：不同类型的单位加载""" self.loader.load_all_units() # 测试步兵 infantry = self.loader.create_unit_from_id("ger_infantry_grenadier") assert infantry.unit_type == UnitType.INFANTRY # 测试坦克 tank = self.loader.create_unit_from_id("ger_tank_panzer_iv") assert tank.unit_type == UnitType.TANK assert "BLITZ" in tank.keywords # 测试火炮 artillery = self.loader.create_unit_from_id("ger_artillery_88mm") assert artillery.unit_type == UnitType.ARTILLERY assert "GUARD" in artillery.keywords # 测试战斗机 fighter = self.loader.create_unit_from_id("ger_fighter_bf109") assert fighter.unit_type == UnitType.FIGHTER # 测试轰炸机 bomber = self.loader.create_unit_from_id("ger_bomber_stuka") assert bomber.unit_type == UnitType.BOMBER def test_units_by_nation(self): """测试：按国家获取单位""" self.loader.load_all_units() german_units = self.loader.get_units_by_nation("GERMANY") usa_units = self.loader.get_units_by_nation("USA") assert len(german_units) > 0 assert len(usa_units) > 0 # 验证国家分类正确 for unit_data in german_units: assert unit_data['nation'] == 'GERMANY' for unit_data in usa_units: assert unit_data['nation'] == 'USA' def test_units_by_type(self): """测试：按类型获取单位""" self.loader.load_all_units() infantry_units = self.loader.get_units_by_type("INFANTRY") tank_units = self.loader.get_units_by_type("TANK") assert len(infantry_units) > 0 assert len(tank_units) > 0 # 验证类型分类正确 for unit_data in infantry_units: assert unit_data['type'].upper() == 'INFANTRY' def test_invalid_unit_id(self): """测试：无效单位ID处理""" self.loader.load_all_units() with pytest.raises(ValueError, match="Unit definition not found"): self.loader.create_unit_from_id("invalid_unit_id") def test_backward_compatibility_activation_cost(self): """测试：activation_cost属性向后兼容性""" unit = load_unit("ger_infantry_grenadier") # 验证可以通过stats.operation_cost访问 assert unit.stats.operation_cost == 1 # 验证新的YAML单位正确设置了operation_cost assert hasattr(unit.stats, 'operation_cost') def test_unit_abilities_loading(self): """测试：单位能力的加载""" self.loader.load_all_units() # 测试带能力的轰炸机 bomber = self.loader.create_unit_from_id("ger_bomber_stuka") assert len(bomber.abilities) > 0, "斯图卡应该有部署能力" # 测试带能力的火炮 artillery = self.loader.create_unit_from_id("ger_artillery_nebelwerfer") assert len(artillery.abilities) > 0, "火箭炮应该有特殊能力" class TestGlobalFunctions: """测试全局便捷函数""" def test_global_load_unit(self): """测试：全局load_unit函数""" unit = load_unit("usa_infantry_gi") assert isinstance(unit, Unit) assert unit.name == "American GI" assert unit.unit_type == UnitType.INFANTRY assert unit.nation == "USA" def test_global_list_all_units(self): """测试：全局list_all_units函数""" units = list_all_units() assert isinstance(units, list) assert len(units) > 0 assert "ger_infantry_grenadier" in units assert "usa_infantry_gi" in units def test_global_loader_singleton(self): """测试：全局加载器单例模式""" loader1 = get_unit_loader() loader2 = get_unit_loader() assert loader1 is loader2, "应该返回同一个单例实例" class TestYAMLIntegration: """测试YAML系统与游戏引擎的集成""" def test_yaml_unit_in_battle_engine(self): """测试：YAML单位在战斗引擎中的使用""" from kards_battle.core.battle_engine import BattleEngine engine = BattleEngine("Germany", "USA", debug_mode=True) # 使用YAML单位 unit = load_unit("ger_infantry_grenadier") # 部署单位 result = engine.deploy_unit_to_support(unit, 0) assert result['success'] # 验证单位正确部署 support_units = engine.battlefield.player1_support.get_all_units() assert len(support_units) == 1 assert support_units[0].name == "German Grenadier" def test_yaml_unit_movement_costs(self): """测试：YAML单位的移动成本""" from kards_battle.core.battle_engine import BattleEngine engine = BattleEngine("Germany", "USA", debug_mode=True) # 使用不同成本的YAML单位 cheap_unit = load_unit("ger_infantry_grenadier") # cost=1 expensive_unit = load_unit("ger_tank_tiger") # cost=4 engine.deploy_unit_to_support(cheap_unit, 0) engine.deploy_unit_to_support(expensive_unit, 0) # 测试便宜单位可以移动 from kards_battle.core.enums import LineType engine.debug_set_kredits(0, kredits=2) result1 = engine.move_unit(cheap_unit.id, (LineType.FRONT, 0), 0) assert result1['success'] # 测试昂贵单位费用不足时不能移动 result2 = engine.move_unit(expensive_unit.id, (LineType.FRONT, 1), 0) assert not result2['success'] assert "Insufficient Kredits" in result2['reason'] if __name__ == "__main__": pytest.main([__file__, "-v"])