From 806e2c1262c8628a53f3409920d4b8a5e0212d6a Mon Sep 17 00:00:00 2001 From: bridge Date: Sun, 4 Jan 2026 21:49:58 +0800 Subject: [PATCH] refactor price system --- src/classes/avatar/inventory_mixin.py | 16 +++---- src/classes/effects.py | 13 ++++++ src/classes/prices.py | 63 +++++++++++++++++++++++++-- src/utils/effect_desc.py | 1 + static/game_configs/auxiliary.csv | 4 +- static/game_configs/persona.csv | 6 +-- static/game_configs/sect.csv | 4 +- tests/test_circulation.py | 14 ++++++ tests/test_prices.py | 41 ++++++++++++----- 9 files changed, 131 insertions(+), 31 deletions(-) diff --git a/src/classes/avatar/inventory_mixin.py b/src/classes/avatar/inventory_mixin.py index 4944f4f..3ed284c 100644 --- a/src/classes/avatar/inventory_mixin.py +++ b/src/classes/avatar/inventory_mixin.py @@ -106,11 +106,6 @@ class InventoryMixin: # ==================== 出售接口 ==================== - def _get_sell_multiplier(self: "Avatar") -> float: - """获取出售价格倍率（包含效果加成）""" - raw = self.effects.get("extra_item_sell_price_multiplier", 0.0) - return 1.0 + float(raw or 0.0) - def sell_item(self: "Avatar", item: "Item", quantity: int = 1) -> int: """ 出售材料物品，返回获得的灵石数量。 @@ -123,8 +118,9 @@ class InventoryMixin: self.remove_item(item, quantity) - base_price = prices.get_item_price(item) * quantity - total = int(base_price * self._get_sell_multiplier()) + # 使用统一的卖出价格接口（包含所有加成逻辑） + unit_price = prices.get_selling_price(item, self) + total = unit_price * quantity self.magic_stone = self.magic_stone + total return total @@ -139,7 +135,8 @@ class InventoryMixin: # 记录流转 self.world.circulation.add_weapon(weapon) - total = int(prices.get_weapon_price(weapon) * self._get_sell_multiplier()) + # 使用统一的卖出价格接口 + total = prices.get_selling_price(weapon, self) self.magic_stone = self.magic_stone + total return total @@ -153,7 +150,8 @@ class InventoryMixin: # 记录流转 self.world.circulation.add_auxiliary(auxiliary) - total = int(prices.get_auxiliary_price(auxiliary) * self._get_sell_multiplier()) + # 使用统一的卖出价格接口 + total = prices.get_selling_price(auxiliary, self) self.magic_stone = self.magic_stone + total return total diff --git a/src/classes/effects.py b/src/classes/effects.py index 9c68e0c..0ab8783 100644 --- a/src/classes/effects.py +++ b/src/classes/effects.py @@ -308,6 +308,18 @@ EXTRA_ITEM_SELL_PRICE_MULTIPLIER = "extra_item_sell_price_multiplier" - 奸商: 0.5 """ +SHOP_BUY_PRICE_REDUCTION = "shop_buy_price_reduction" +""" +商铺购买价格倍率减免 +类型: float +结算: src/classes/prices.py +说明: 降低从系统购买物品时的溢价倍率。 +数值参考: + - 微量: 0.1 (倍率-0.1) + - 中量: 0.5 (倍率-0.5) +限制: 最终倍率最低为 1.0 +""" + EXTRA_PLUNDER_MULTIPLIER = "extra_plunder_multiplier" """ 额外搜刮收益倍率 @@ -429,6 +441,7 @@ ALL_EFFECTS = [ # 经济相关 "extra_item_sell_price_multiplier", # float - 额外物品出售价格倍率 + "shop_buy_price_reduction", # float - 商铺购买价格倍率减免 "extra_plunder_multiplier", # float - 额外搜刮收益倍率 # 特殊权限 diff --git a/src/classes/prices.py b/src/classes/prices.py index 53abf2f..b226d29 100644 --- a/src/classes/prices.py +++ b/src/classes/prices.py @@ -20,6 +20,7 @@ if TYPE_CHECKING: from src.classes.item import Item from src.classes.weapon import Weapon from src.classes.auxiliary import Auxiliary + from src.classes.avatar import Avatar # 类型别名 Sellable = Union["Item", "Weapon", "Auxiliary"] @@ -31,6 +32,9 @@ class Prices: 所有城镇可交易物品/装备的价格在此统一管理。 """ + # 全局购买倍率（玩家从系统购买时的溢价） + GLOBAL_BUY_MULTIPLIER = 1.5 + # 材料价格表（采集物等） ITEM_PRICES = { Realm.Qi_Refinement: 10, @@ -56,21 +60,22 @@ class Prices: } def get_item_price(self, item: "Item") -> int: - """获取材料价格""" + """获取材料基础价格""" return self.ITEM_PRICES.get(item.realm, 10) def get_weapon_price(self, weapon: "Weapon") -> int: - """获取兵器价格""" + """获取兵器基础价格""" return self.WEAPON_PRICES.get(weapon.realm, 100) def get_auxiliary_price(self, auxiliary: "Auxiliary") -> int: - """获取辅助装备价格""" + """获取辅助装备基础价格""" return self.AUXILIARY_PRICES.get(auxiliary.realm, 80) def get_price(self, obj: Sellable) -> int: """ - 统一价格查询接口。 + 统一基础价格查询接口。 根据对象类型自动分发到对应的价格查询方法。 + 注意：这是物品的【基准价值】，通常等于玩家【卖出给系统】的基础价格。 """ from src.classes.item import Item from src.classes.weapon import Weapon @@ -84,6 +89,56 @@ class Prices: return self.get_auxiliary_price(obj) return 0 + def get_buying_price(self, obj: Sellable, buyer: "Avatar" = None) -> int: + """ + 获取玩家购买价格（从系统商店购买）。 + + 计算公式： + 基础价格 * max(1.0, GLOBAL_BUY_MULTIPLIER - 折扣) + + Args: + obj: 交易物品 + buyer: 买家（用于计算折扣） + """ + base_price = self.get_price(obj) + + # 默认倍率 1.5 + multiplier = self.GLOBAL_BUY_MULTIPLIER + + if buyer is not None: + # 获取折扣 (倍率减免) + # 例如 0.1 表示倍率 -0.1 + reduction = float(buyer.effects.get("shop_buy_price_reduction", 0.0)) + multiplier -= reduction + + # 保证倍率不低于 1.0 (原价) + final_multiplier = max(1.0, multiplier) + + return int(base_price * final_multiplier) + + def get_selling_price(self, obj: Sellable, seller: "Avatar" = None) -> int: + """ + 获取玩家卖出价格（卖给系统商店）。 + + 计算公式： + 基础价格 * (1.0 + 卖出加成) + + Args: + obj: 交易物品 + seller: 卖家（用于计算加成） + """ + base_price = self.get_price(obj) + + multiplier = 1.0 + + if seller is not None: + # 获取卖出加成 + # 例如 0.2 表示价格 +20% + bonus = float(seller.effects.get("extra_item_sell_price_multiplier", 0.0)) + multiplier += bonus + + return int(base_price * multiplier) + # 全局单例 prices = Prices() diff --git a/src/utils/effect_desc.py b/src/utils/effect_desc.py index 4b09267..676c489 100644 --- a/src/utils/effect_desc.py +++ b/src/utils/effect_desc.py @@ -13,6 +13,7 @@ EFFECT_DESC_MAP = { "extra_harvest_items": "采集获取物品", "extra_hunt_items": "狩猎获取物品", "extra_item_sell_price_multiplier": "物品出售价格", + "shop_buy_price_reduction": "购买折扣", "extra_weapon_upgrade_chance": "兵器升级概率", "extra_plunder_multiplier": "搜刮收益", "extra_catch_success_rate": "捕捉灵兽成功率", diff --git a/static/game_configs/auxiliary.csv b/static/game_configs/auxiliary.csv index bb4e9de..4899b3a 100644 --- a/static/game_configs/auxiliary.csv +++ b/static/game_configs/auxiliary.csv @@ -3,7 +3,7 @@ id,name,grade,desc,effects 2001,木灵甲,筑基,以灵木纤维编织，生机盎然，防御尚可.,{damage_reduction: 0.08} 2002,血菩提,筑基,生于火麒麟滴血之处，重伤必治，无伤增功.,{extra_hp_recovery_rate: 0.3} 2003,人参果(伪),筑基,虽不及真品，闻一闻也能活三百六十岁.,{extra_max_lifespan: 50} -2004,万宝令,筑基,千帆城通用的最高级商令，见令如见城主.,{extra_item_sell_price_multiplier: 0.2} +2004,万宝令,筑基,千帆城通用的高级商令。,"{extra_item_sell_price_multiplier: 0.1, shop_buy_price_reduction: 0.1}" 2005,铁精,筑基,提炼百斤凡铁才得一两，提升法器品质的基础材料.,{extra_weapon_upgrade_chance: 0.05} 2006,源天神眼,筑基,可勘破虚妄，直视本源，寻龙定穴.,"{extra_observation_radius: 1, extra_fortune_probability: 0.005}" 2007,踏云靴,筑基,步履生云，身轻如燕，赶路必备.,"{extra_escape_success_rate: 0.15, extra_move_step: 1}" @@ -19,7 +19,7 @@ id,name,grade,desc,effects 3006,御兽环,金丹,套在妖兽颈上，任其凶焰滔天，也要乖乖听命.,{extra_catch_success_rate: 0.1} 3007,太虚甲,金丹,太虚幻境所化，万法不沾，防御力惊人.,"{damage_reduction: 0.2, extra_max_hp: 200}" 3008,八卦炉,金丹,仿太上老君炼丹炉，丹成九转，药效倍增.,"{extra_hp_recovery_rate: 0.5, extra_cast_success_rate: 0.2}" -3009,聚宝盆,金丹,沈万三的宝物，放入一文，取出万贯，财源滚滚.,"{extra_item_sell_price_multiplier: 0.5, extra_fortune_probability: 0.01}" +3009,聚宝盆,金丹,放入一文，取出万贯，财源滚滚.,"{extra_item_sell_price_multiplier: 0.2, extra_fortune_probability: 0.01, shop_buy_price_reduction: 0.2}" 3010,天地烘炉,金丹,以天地为炉，造化为工，炼制万物.,"{extra_weapon_upgrade_chance: 0.15, extra_cast_success_rate: 0.3}" 3011,六道剑匣,金丹,背负六道，剑出轮回，悟剑良伴.,"{extra_breakthrough_success_rate: 0.2, extra_cultivate_exp: 100}" 3012,传国玉玺,元婴,受命于天，既寿永昌。皇威浩荡，震慑宵小.,{realm_suppression_bonus: 0.15} diff --git a/static/game_configs/persona.csv b/static/game_configs/persona.csv index 4d50975..5441271 100644 --- a/static/game_configs/persona.csv +++ b/static/game_configs/persona.csv @@ -5,7 +5,7 @@ id,name,exclusion_names,desc,rarity,condition,effects 3,怠惰,冒险;极端正义;极端邪恶,你总是会拖延，不想努力，更热衷于享受人生。,N,,{extra_cast_success_rate: -0.05} 4,冒险,怠惰;惜命,你总是会冒险，喜欢刺激，总想放手一搏。,N, 5,随性,理性;极端正义;极端邪恶,你总是会随机应变，性子到哪里了就是哪里，没有一定之规。,N, -6,贪财,,你对灵石和财富有着强烈的渴望。,N, +6,贪财,,你对灵石和财富有着强烈的渴望。,N,,{extra_item_sell_price_multiplier: 0.1} 7,采集者,,喜欢在山林中寻找各种奇花异草和灵药，对植物有着敏锐的直觉和深厚的兴趣。你认为大自然的恩赐需要用心去发现和珍惜。,R,,{extra_harvest_items: 1} 8,猎人,,享受在野外追踪猎物的刺激感，对各种动物的习性了如指掌，喜欢捕猎野兽。情况允许也会御兽。,R,,{extra_hunt_items: 1} 9,沉思,无常,你总是会深思熟虑，思考问题比较有哲理。,N, @@ -39,7 +39,7 @@ id,name,exclusion_names,desc,rarity,condition,effects 37,疑心重,热情;外向;友爱,对任何人都保持怀疑，不轻易信任，时刻警惕他人可能的背叛和算计。,N, 38,体修,惜命;胆小,专注肉身淬炼，体魄强健，近战无敌。你相信肉身才是修行的根本，法术只是旁门左道。,R,,"[{when: 'avatar.weapon is None', extra_battle_strength_points: 5}]" 39,炼丹师,好斗,精通丹道，对灵药敏感，擅长炼制丹药。你认为丹药是修行的关键，战斗并非你的专长。,R, -40,福缘深厚,,天生福运，逢凶化吉。虽不如气运之子，但也常有小幸运眷顾。,SR,,{extra_fortune_probability: 0.01} +40,福缘深厚,,天生福运，逢凶化吉。虽不如气运之子，但也常有小幸运眷顾。,SR,,"{extra_fortune_probability: 0.01, shop_buy_price_reduction: 0.1}" 41,剑修,怠惰,对剑道有着深厚的兴趣和独特的天赋，持剑修炼时事半功倍。,R,,"[{when: 'avatar.weapon.type == WeaponType.SWORD', extra_weapon_proficiency_gain: 0.5, extra_battle_strength_points: 1}]" 42,刀修,怠惰,专精刀法，对刀道有着独特的理解，持刀修炼时事半功倍。,R,,"[{when: 'avatar.weapon.type == WeaponType.SABER', extra_weapon_proficiency_gain: 0.5, extra_battle_strength_points: 1}]" 43,枪修,怠惰,精通枪法，百兵之王在手，修炼速度远超常人。,R,,"[{when: 'avatar.weapon.type == WeaponType.SPEAR', extra_weapon_proficiency_gain: 0.5, extra_battle_strength_points: 1}]" @@ -58,6 +58,6 @@ id,name,exclusion_names,desc,rarity,condition,effects 56,无欲,好色,你清心寡欲，对男女之事毫无兴趣，面对美色心如止水，不为所动。,N,, 57,探险家,死宅,你对未知的世界充满好奇，热衷于探索地图上未被点亮的区域，难以忍受长时间待在同一个地方。,SR,, 58,死宅,探险家;冒险,你认为外面的世界充满危险且麻烦，除了必要的宗门任务，你极度排斥长途跋涉，更喜欢待在熟悉的区域修炼。,N,, -59,扫把星,福缘深厚;气运之子,天生霉运缠身，喝凉水都塞牙。虽然活着不易，但也磨练了心性。,N,,{extra_misfortune_probability: 0.005} +59,扫把星,福缘深厚;气运之子,天生霉运缠身，喝凉水都塞牙。虽然活着不易，但也磨练了心性。,N,,"{extra_misfortune_probability: 0.005, extra_item_sell_price_multiplier: -0.1}" 60,大器晚成,,早年修行多舛，霉运连连；但若能坚持至金丹元婴，便可否极泰来，气运亨通。,SR,,"[{when: 'avatar.cultivation_progress.realm.value in [""练气"", ""筑基""]', extra_misfortune_probability: 0.005}, {when: 'avatar.cultivation_progress.realm.value in [""金丹"", ""元婴""]', extra_fortune_probability: 0.01}]" 61,炼器师,好斗,精通炼器之道，对材料敏锐，擅长铸造法宝。你认为法宝是修行的关键，战斗并非你的专长。,R,,{extra_cast_success_rate: 0.15} diff --git a/static/game_configs/sect.csv b/static/game_configs/sect.csv index c5045ba..7768bda 100644 --- a/static/game_configs/sect.csv +++ b/static/game_configs/sect.csv @@ -5,10 +5,10 @@ id,name,desc,member_act_style,alignment,weight,preferred_weapon,effects,rank_nam 3,水镜宗,"正道十宗之一，实则严守中立。拥有仙界异宝""彻天水镜""可预知未来。【趋吉避凶】你拥有超乎常人的直觉，视野开阔，且极易在探索中发现奇遇。",处事冷静圆融，喜以柔克刚，擅借力与反制。,中立,1,扇,"{extra_observation_radius: 2, extra_fortune_probability: 0.002}",镜主;掌镜人;传人;侍镜 4,冥王宗,"行走幽冥之道，术法阴冷狠厉。【通幽】你修行幽冥之法，心志坚定，突破瓶颈时心无杂念，成功率更高。",言辞冷厉少情，敬畏因果而不惧杀伐，偏向效率与结果。,邪,1,扇,"{extra_breakthrough_success_rate: 0.1}",殿主;判官;无常;鬼卒 5,朱勾宗,"邪宗大派。以炼器、机关、暗杀闻名于世，素来阴毒冷僻。【暗杀专家】你精通潜伏与刺杀，对于强敌，避开正面交锋、伺机暗杀往往是你最佳的制胜之道。",直面欲望与代价，不惧黑暗，以攻伐见长。,邪,1,暗器,"{extra_assassinate_success_rate: 0.15, extra_battle_strength_points: 1, extra_cast_success_rate: 0.05}",楼主;掌刑使;影刺;探子 -6,合欢宗,"以情入道，靠双修增进修为，善驭人心，长于权变。【双修秘术】你的体质特殊，通过双修能获得远超常人的修为提升，这是你提升境界的最佳捷径。",辞令婉转，善于拿捏人欲与局势，以柔制刚。,中立,1,琴,"{extra_dual_cultivation_exp: 150}",宫主;护法;媚仙;侍童 +6,合欢宗,"以情入道，靠双修增进修为，善驭人心，长于权变。【双修秘术】你的体质特殊，通过双修能获得远超常人的修为提升，这是你提升境界的最佳捷径。",辞令婉转，善于拿捏人欲与局势，以柔制刚。,中立,1,琴,"{extra_dual_cultivation_exp: 150, shop_buy_price_reduction: 0.1}",宫主;护法;媚仙;侍童 7,镇魂宗,"铁血风格，擅安魂、封邪、渡厄，兼有刚烈镇压之术。【金刚不坏】你修习护体神功，拥有极强的肉身防御与伤害减免能力，如山岳般不可撼动。",肃穆沉稳，重安魂镇邪，少言而果决。,正,1,棍,"{extra_max_hp: 100, damage_reduction: 0.15}",住持;首座;行者;居士 8,幽魂噬影宗,"镇宗典籍《幽冥录》。幽明气为根基。【如影随形】你极擅身法与遁术，遇到危险时总能全身而退，亦能伺机刺杀落单之敌。",行事隐秘果断，重结果轻虚名，擅潜行与出其不意。,邪,1,刀,"{extra_escape_success_rate: 0.4, extra_assassinate_success_rate: 0.15}",门主;护法;影卫;探子 -9,千帆城,"炼器大宗，巧匠云集。著名法宝有灵灭丝、定魂蓝星等。商旅云集，自成体系。【巧夺天工】你深谙炼器与经商之道，温养兵器时常能使其脱胎换骨，且在交易中总能获利。",务实精明，重交易与信誉，崇尚规则与秩序。,中立,1,枪,"{extra_item_sell_price_multiplier: 0.2, extra_weapon_upgrade_chance: 0.15, extra_cast_success_rate: 0.05}",城主;大供奉;执事;学徒 +9,千帆城,"炼器大宗，巧匠云集。著名法宝有灵灭丝、定魂蓝星等。商旅云集，自成体系。【巧夺天工】你深谙炼器与经商之道，温养兵器时常能使其脱胎换骨，且在交易中总能获利。",务实精明，重交易与信誉，崇尚规则与秩序。,中立,1,枪,"{extra_item_sell_price_multiplier: 0.05, extra_weapon_upgrade_chance: 0.15, extra_cast_success_rate: 0.05, shop_buy_price_reduction: 0.05}",城主;大供奉;执事;学徒 10,妙化宗,"精擅音律杀伐与精神操控，功法诡谲阴柔，以无形琴音乱人心智，杀人于无形。【魔音贯耳】你擅长以音律乱人心智，灵力深厚绵长，虽不擅肉搏，但手段诡谲。",文雅缥缈中深藏算计，喜用言语与音律掌控局势，杀伐不沾烟火气。,邪,1,琴,"{extra_escape_success_rate: 0.5, extra_misfortune_probability: -0.005}", 11,回玄宗,"当世第一阵法大派，以禁制阵诀独步天下，讲究阵法推演与巧思妙用。【阵法聚灵】你善用阵法辅助修炼，能够聚集天地灵气，修炼速度快于常人，且常有意外收获。",严谨细致，追求技术极致，战斗中擅长以后手禁制反制敌人，步步为营。,正,1,暗器,"{cultivate_duration_reduction: 0.1, extra_fortune_probability: 0.005}", 12,不夜城,"修习极光玄真法，以万里极光壁的绝对防御闻名于世。【生生不息】你的生命力极其顽强，疗伤效果倍增，寿元亦远超同阶修士。",坚韧顽强，在逆境中图强，行事光明磊落但也深谋远虑，极其护短。,正,1,扇,"{extra_hp_recovery_rate: 0.5, extra_max_lifespan: 20}",城主;阁老;执令;守夜人 diff --git a/tests/test_circulation.py b/tests/test_circulation.py index 59740f3..1298924 100644 --- a/tests/test_circulation.py +++ b/tests/test_circulation.py @@ -157,7 +157,21 @@ def test_avatar_sell_integration(empty_world): with patch("src.classes.prices.prices") as mock_prices: mock_prices.get_weapon_price.return_value = 100 mock_prices.get_auxiliary_price.return_value = 200 + # 这里的 sell_weapon 现在调用 get_selling_price，我们也 mock 它 + # 假设这里没有额外加成，或者我们直接设定最终售价 + mock_prices.get_selling_price.side_effect = lambda obj, seller: \ + 100 if getattr(obj, "id", 0) == 999 else \ + (300 if getattr(obj, "id", 0) == 888 else 0) + # 上面的 side_effect 比较复杂，因为测试里先后卖了 weapon(id=999) 和 aux(id=888) + # 我们可以简单地根据类型返回，或者分段 mock + # 重新定义 mock 逻辑 + def get_selling_price_mock(obj, seller): + if hasattr(obj, "id") and obj.id == 999: return 100 + if hasattr(obj, "id") and obj.id == 888: return 200 + return 0 + mock_prices.get_selling_price.side_effect = get_selling_price_mock + # 1. Test Sell Weapon # Create a dummy weapon that acts like the real one weapon = MagicMock(spec=Weapon) diff --git a/tests/test_prices.py b/tests/test_prices.py index d28d888..78f6157 100644 --- a/tests/test_prices.py +++ b/tests/test_prices.py @@ -152,13 +152,30 @@ class TestAvatarSell: base_price = prices.get_item_price(item) - # 设置 20% 加成 - patch 内部方法 - with patch.object(dummy_avatar, '_get_sell_multiplier', return_value=1.2): - gained = dummy_avatar.sell_item(item, 1) + # 模拟 20% 加成 (0.2) + # 这里的 effects 是 property，需要用 PropertyMock + with patch("src.classes.avatar.core.Avatar.effects", new_callable=lambda: {"extra_item_sell_price_multiplier": 0.2}): + # 注意：由于 Avatar 分布在多个 mixin 中，patch 的位置取决于 effects 定义的位置 + # effects 定义在 EffectsMixin 中，但混入后是在 Avatar 类上 + # 如果 patch 比较麻烦，我们可以利用 Prices.get_selling_price 的逻辑 + # 这里我们其实也可以直接 patch get_selling_price 来验证 sell_item 是否使用了它 + # 但为了验证集成逻辑，我们尝试 patch effects + pass + + # 重新考虑：patch Property 比较繁琐，容易出错。 + # 既然我们已经验证了 Prices.get_selling_price 的逻辑（在 Prices 单元测试中应该覆盖，虽然当前文件主要是测试 Avatar 交互） + # 这里我们可以构造一个带有特定 effects 的 mock avatar，或者更简单的： + # patch prices.get_selling_price，验证其被调用，且返回值被正确累加 - expected = int(base_price * 1.2) - assert gained == expected - assert dummy_avatar.magic_stone.value == expected + expected_total = int(base_price * 1.2) + with patch("src.classes.prices.prices.get_selling_price", return_value=expected_total) as mock_get_price: + gained = dummy_avatar.sell_item(item, 1) + + # 验证调用参数 + mock_get_price.assert_called_with(item, dummy_avatar) + + assert gained == expected_total + assert dummy_avatar.magic_stone.value == expected_total def test_sell_weapon_with_multiplier(self, dummy_avatar): """测试出售兵器时价格倍率生效""" @@ -169,10 +186,12 @@ class TestAvatarSell: dummy_avatar.magic_stone.value = 0 base_price = prices.get_weapon_price(weapon) - # 设置 50% 加成 - patch 内部方法 - with patch.object(dummy_avatar, '_get_sell_multiplier', return_value=1.5): - gained = dummy_avatar.sell_weapon(weapon) + expected_total = int(base_price * 1.5) - expected = int(base_price * 1.5) - assert gained == expected + # 同样 patch get_selling_price + with patch("src.classes.prices.prices.get_selling_price", return_value=expected_total) as mock_get_price: + gained = dummy_avatar.sell_weapon(weapon) + mock_get_price.assert_called_with(weapon, dummy_avatar) + + assert gained == expected_total