为何@mock.patch无法成功模拟我的TheFlop类?
解决@mock.patch未成功模拟TheFlop类的问题
一、先验证mock路径是否正确
mock.patch的路径必须是被测试代码实际导入TheFlop的路径,不是TheFlop所在的原模块路径,这是最常见的踩坑点:
- 举个例子:如果你的
AnalyseMyHandOnTurn类在my_hand_analysis.py文件里,并且是通过from flop_turn_river_cards import TheFlop导入的,那mock的路径就应该是my_hand_analysis.TheFlop,而不是你现在用的flop_turn_river_cards.TheFlop - 验证方法:直接看
AnalyseMyHandOnTurn所在文件的导入语句,确认TheFlop是从哪个路径导入的;或者在测试函数里加一行print(AnalyseMyHandOnTurn.__module__),查看被测试类所属模块,再对应到导入路径。
二、路径正确后,确保模拟生效的步骤
- 验证mock类是否被调用:在测试里加断言,确认TheFlop的mock类确实被实例化了,比如:
如果这个断言失败,说明被测试方法里根本没用到TheFlop类,或者路径还是错的。mockTheFlop.assert_called() # 或者assert_called_once(),根据实际调用次数调整 - 设置mock实例的方法返回值:如果
is_board_paired_on_turn里会调用TheFlop实例的方法,得给这些方法设置预期返回值,否则mock实例的方法会默认返回MagicMock对象,可能导致逻辑不符合预期:# 获取mock类的实例对象 mock_flop_instance = mockTheFlop.return_value # 比如TheFlop有个check_pair方法,设置它返回True mock_flop_instance.check_pair.return_value = True - 检查被测试方法的逻辑:确认
is_board_paired_on_turn里确实是通过导入的TheFlop类来实例化,而不是直接在方法里写死了其他实现。
三、修正后的测试代码示例
假设AnalyseMyHandOnTurn所在模块是my_hand_analysis,正确的测试代码如下:
@mock.patch('my_hand_analysis.TheFlop') def test_is_board_paired_on_turn(self, mockTheFlop): # 配置mock实例的方法返回值 mock_flop = mockTheFlop.return_value mock_flop.check_pair.return_value = True # 根据实际业务逻辑调整 amhot = AnalyseMyHandOnTurn() amhot.turn1 = [[10, 'S'], [10, 'C'],[7, 'C'], [2, 'S']] amhot.turn2 = [[10, 'S'], [10, 'C'],[7, 'C'], [2, 'S']] amhot.did_board_pair_on_turn1 = True amhot.did_board_pair_on_turn2 = True amhot.board_pair_top_card_turn1 = True amhot.board_pair_top_card_turn2 = True amhot.is_board_paired_on_turn() # 验证mock类是否被调用 mockTheFlop.assert_called() # 执行原有断言 assert amhot.hand_ratings_turn1['board_paired'] == 6 assert amhot.hand_ratings_turn2['board_paired'] == 6
内容的提问来源于stack exchange,提问作者dzdsds
相关产品推荐
相关产品推荐

