fix

tests/test_task3.py

@@ -0,0 +1,100 @@
+"""测试任务3 - 使用poker_task1进行扑克牌EMD距离计算"""
+import sys
+import os
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+# 从poker_task3.task3模块导入函数
+from poker_task3.task3 import parseTurnInput, calHandEquityHist, calPokerEmdTurn
+from poker_task3.card import Card
+
+
+def test_parse_turn_input():
+    """测试TURN阶段输入解析"""
+    input_text = """
+    As Ks
+    7s 6s
+    8h 9d 9c Qh
+    """
+    
+    hand1, hand2, board = parseTurnInput(input_text)
+    
+    assert len(hand1) == 2
+    assert len(hand2) == 2
+    assert len(board) == 4
+    
+    assert str(hand1[0]) == "As"
+    assert str(hand1[1]) == "Ks"
+    assert str(hand2[0]) == "7s"
+    assert str(hand2[1]) == "6s"
+    
+    print("✓ TURN输入解析测试通过")
+
+
+def test_hand_equity_Hist():
+    """测试手牌胜率分布计算"""
+    hole_cards = Card.parse_cards("As Ks")
+    board_cards = Card.parse_cards("8h 9d 9c Qh")
+    
+    Hist = calHandEquityHist(hole_cards, board_cards)
+    
+    assert len(Hist) == 30  # 胜率分布的区间数量（改进为30个区间）
+    assert all(isinstance(x, (int, float)) for x in Hist)
+    assert all(x >= 0 for x in Hist)  # 所有值都应该是非负的
+    assert abs(sum(Hist) - 1.0) < 1e-10  # 分布总和应该为1（标准化后）
+    
+    print("✓ 手牌胜率分布测试通过")
+
+
+def test_poker_emd_calculation():
+    """测试主要的EMD计算函数"""
+    input_text = """As Ks
+7s 6s
+8h 9d 9c Qh"""
+    
+    distance = calPokerEmdTurn(input_text)
+    
+    assert isinstance(distance, (int, float))
+    assert distance >= 0  # EMD总是非负的
+    
+    print(f"✓ 扑克牌EMD计算测试通过 (距离: {distance:.3f})")
+
+
+def test_different_hand_strengths():
+    """测试不同强度手牌的EMD"""
+    # 强牌 vs 弱牌
+    input_text = """As Ks
+2c 3d
+8h 9d 9c Qh"""
+    
+    distance = calPokerEmdTurn(input_text)
+    
+    assert isinstance(distance, (int, float))
+    assert distance >= 0  # 应该有非负距离
+    
+    print(f"✓ 不同手牌强度测试通过 (距离: {distance:.3f})")
+
+
+def test_error_handling():
+    """测试无效输入的错误处理"""
+    # 错误的行数
+    result = parseTurnInput("As Ks\n7s 6s")
+    assert result == (None, None, None), "应该返回(None, None, None)"
+    
+    # 手牌中牌数错误
+    result = parseTurnInput("As Ks Qs\n7s 6s\n8h 9d 9c Qh")
+    assert result == (None, None, None), "应该返回(None, None, None)"
+    
+    # 公共牌数错误
+    result = parseTurnInput("As Ks\n7s 6s\n8h 9d 9c")
+    assert result == (None, None, None), "应该返回(None, None, None)"
+    
+    print("✓ 错误处理测试通过")
+
+
+if __name__ == "__main__":
+    test_parse_turn_input()
+    test_hand_equity_Hist()
+    test_poker_emd_calculation()
+    test_different_hand_strengths()
+    test_error_handling()
+    print("所有任务3测试通过! ✓")