125 lines
4.4 KiB
Python
125 lines
4.4 KiB
Python
import pytest
|
||
from src.classes.weapon import Weapon, get_random_weapon_by_realm, weapons_by_id
|
||
from src.classes.auxiliary import Auxiliary, get_random_auxiliary_by_realm, auxiliaries_by_id
|
||
from src.classes.cultivation import Realm
|
||
from src.classes.weapon_type import WeaponType
|
||
|
||
class TestEquipment:
|
||
def test_weapon_structure(self):
|
||
"""测试兵器数据结构和加载"""
|
||
assert len(weapons_by_id) > 0
|
||
|
||
# 检查任意一个兵器
|
||
weapon = next(iter(weapons_by_id.values()))
|
||
assert isinstance(weapon, Weapon)
|
||
assert isinstance(weapon.realm, Realm)
|
||
assert isinstance(weapon.weapon_type, WeaponType)
|
||
assert weapon.name
|
||
|
||
# 测试 info 方法
|
||
info = weapon.get_info()
|
||
assert info == weapon.name
|
||
|
||
detailed = weapon.get_detailed_info()
|
||
assert weapon.name in detailed
|
||
assert weapon.realm.value in detailed
|
||
|
||
def test_weapon_random_generation(self):
|
||
"""测试按境界随机生成兵器"""
|
||
# 测试练气期
|
||
w_qi = get_random_weapon_by_realm(Realm.Qi_Refinement)
|
||
if w_qi:
|
||
assert w_qi.realm == Realm.Qi_Refinement
|
||
|
||
# 测试筑基期
|
||
w_found = get_random_weapon_by_realm(Realm.Foundation_Establishment)
|
||
if w_found:
|
||
assert w_found.realm == Realm.Foundation_Establishment
|
||
|
||
# 测试金丹期
|
||
w_core = get_random_weapon_by_realm(Realm.Core_Formation)
|
||
if w_core:
|
||
assert w_core.realm == Realm.Core_Formation
|
||
|
||
def test_weapon_deepcopy(self):
|
||
"""测试兵器深拷贝独立性"""
|
||
w1 = get_random_weapon_by_realm(Realm.Qi_Refinement)
|
||
if not w1:
|
||
pytest.skip("No Qi Refinement weapons available")
|
||
|
||
w2 = get_random_weapon_by_realm(Realm.Qi_Refinement)
|
||
# 即使随机到同一个原型，它们也应该是不同的对象
|
||
# 注意：get_random_weapon_by_realm 内部已经做了 deepcopy
|
||
|
||
# 为了确保测试有效，我们手动获取同一个原型并 deepcopy
|
||
prototype = weapons_by_id[w1.id]
|
||
import copy
|
||
w_copy1 = copy.deepcopy(prototype)
|
||
w_copy2 = copy.deepcopy(prototype)
|
||
|
||
assert w_copy1 is not w_copy2
|
||
assert w_copy1.id == w_copy2.id
|
||
|
||
# 修改一个不影响另一个
|
||
w_copy1.special_data["test"] = 123
|
||
assert "test" not in w_copy2.special_data
|
||
|
||
def test_auxiliary_structure(self):
|
||
"""测试辅助装备数据结构和加载"""
|
||
assert len(auxiliaries_by_id) > 0
|
||
|
||
# 检查任意一个
|
||
aux = next(iter(auxiliaries_by_id.values()))
|
||
assert isinstance(aux, Auxiliary)
|
||
assert isinstance(aux.realm, Realm)
|
||
assert aux.name
|
||
|
||
# 测试 info 方法
|
||
info = aux.get_info()
|
||
assert info == aux.name
|
||
|
||
detailed = aux.get_detailed_info()
|
||
assert aux.name in detailed
|
||
assert aux.realm.value in detailed
|
||
|
||
def test_soul_banner(self):
|
||
"""测试万魂幡特殊逻辑"""
|
||
# 查找万魂幡
|
||
soul_banner = None
|
||
for a in auxiliaries_by_id.values():
|
||
if a.name == "万魂幡":
|
||
soul_banner = a
|
||
break
|
||
|
||
if not soul_banner:
|
||
pytest.skip("万魂幡 not found in config")
|
||
|
||
import copy
|
||
banner = copy.deepcopy(soul_banner)
|
||
|
||
# 初始状态
|
||
assert "吞噬魂魄" not in banner.get_info()
|
||
|
||
# 增加魂魄
|
||
banner.special_data["devoured_souls"] = 100
|
||
|
||
# 检查显示更新
|
||
assert "吞噬魂魄：100" in banner.get_info()
|
||
assert "吞噬魂魄：100" in banner.get_detailed_info()
|
||
|
||
# 检查结构化信息
|
||
struct = banner.get_structured_info()
|
||
assert "已吞噬魂魄：100" in struct["desc"]
|
||
|
||
def test_grade_renaming_compatibility(self):
|
||
"""测试 realm 改名后的兼容性（如果有必要）"""
|
||
# 确保 weapon 和 auxiliary 确实有 realm 属性
|
||
weapon = next(iter(weapons_by_id.values()))
|
||
assert hasattr(weapon, 'realm')
|
||
assert not hasattr(weapon, 'grade') # 确保改名彻底
|
||
|
||
aux = next(iter(auxiliaries_by_id.values()))
|
||
assert hasattr(aux, 'realm')
|
||
assert not hasattr(aux, 'grade') # 确保改名彻底
|
||
|